Cleanup

frigate/track/norfair_tracker.py

@@ -5,7 +5,6 @@ from typing import Any, Sequence, cast
 
 import cv2
 import numpy as np
-from frigate.track.stationary_classifier import StationaryMotionClassifier
 from norfair.drawing.draw_boxes import draw_boxes
 from norfair.drawing.drawer import Drawable, Drawer
 from norfair.filter import OptimizedKalmanFilterFactory
@@ -13,12 +12,12 @@ from norfair.tracker import Detection, TrackedObject, Tracker
 from rich import print
 from rich.console import Console
 from rich.table import Table
-from scipy.ndimage import gaussian_filter
 
 from frigate.camera import PTZMetrics
 from frigate.config import CameraConfig
 from frigate.ptz.autotrack import PtzMotionEstimator
 from frigate.track import ObjectTracker
+from frigate.track.stationary_classifier import StationaryMotionClassifier
 from frigate.util.image import (
     SharedMemoryFrameManager,
     get_histogram,

frigate/track/stationary_classifier.py

@@ -80,7 +80,7 @@ class StationaryMotionClassifier:
             y1,
             x2,
             y2,
-            crop.shape if 'crop' in locals() else None,
+            crop.shape if "crop" in locals() else None,
         )
         return result
 
@@ -115,9 +115,7 @@ class StationaryMotionClassifier:
         """
 
         if id not in self.anchor_crops or id not in self.anchor_boxes:
-            logger.debug(
-                "evaluate: id=%s has no anchor; default keep stationary", id
-            )
+            logger.debug("evaluate: id=%s has no anchor; default keep stationary", id)
             return True
 
         # Compare same spatial region across frames

notebooks/YOLO_NAS_Pretrained_Export.ipynb

@@ -48,13 +48,14 @@
    "source": [
     "# export the model for compatibility with Frigate\n",
     "\n",
-        "model.export(\"yolo_nas_s.onnx\",\n",
+    "model.export(\n",
+    "    \"yolo_nas_s.onnx\",\n",
     "    output_predictions_format=DetectionOutputFormatMode.FLAT_FORMAT,\n",
     "    max_predictions_per_image=20,\n",
     "    num_pre_nms_predictions=300,\n",
     "    confidence_threshold=0.4,\n",
-        "             input_image_shape=(320,320),\n",
-        "            )"
+    "    input_image_shape=(320, 320),\n",
+    ")"
    ]
   },
   {
@@ -67,7 +68,7 @@
    "source": [
     "from google.colab import files\n",
     "\n",
-        "files.download('yolo_nas_s.onnx')"
+    "files.download(\"yolo_nas_s.onnx\")"
    ]
   }
  ],