diff --git a/frigate/detectors/__init__.py b/frigate/detectors/__init__.py index f5c1fb219..f94815f24 100644 --- a/frigate/detectors/__init__.py +++ b/frigate/detectors/__init__.py @@ -21,7 +21,7 @@ def create_detector(det_type: DetectorTypeEnum, **kwargs): _api_types = { DetectorTypeEnum.cpu: CpuTfl, DetectorTypeEnum.edgetpu: EdgeTpuTfl, - DetectorTypeEnum.openvino: OvDetector + DetectorTypeEnum.openvino: OvDetector, } if det_type == DetectorTypeEnum.cpu: diff --git a/frigate/object_detection.py b/frigate/object_detection.py index d67417fb6..b1655d331 100644 --- a/frigate/object_detection.py +++ b/frigate/object_detection.py @@ -52,7 +52,12 @@ class LocalObjectDetector(ObjectDetector): else: self.input_transform = None - self.detect_api = create_detector(det_type, det_device=det_device, model_config=model_config, num_threads=num_threads) + self.detect_api = create_detector( + det_type, + det_device=det_device, + model_config=model_config, + num_threads=num_threads, + ) def detect(self, tensor_input, threshold=0.4): detections = []