administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 09:45:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fixed code formatting

Browse Source
This commit is contained in:
Dennis George 2022-12-08 11:48:31 -06:00
parent 3fa174eca4
commit 3f524da4ca
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/detectors/__init__.py
View File

@ -21,7 +21,7 @@ def create_detector(det_type: DetectorTypeEnum, **kwargs):
_api_types = { _api_types = {
DetectorTypeEnum.cpu: CpuTfl, DetectorTypeEnum.cpu: CpuTfl,
DetectorTypeEnum.edgetpu: EdgeTpuTfl, DetectorTypeEnum.edgetpu: EdgeTpuTfl,
DetectorTypeEnum.openvino: OvDetector DetectorTypeEnum.openvino: OvDetector,
} }
if det_type == DetectorTypeEnum.cpu: if det_type == DetectorTypeEnum.cpu:

7
frigate/object_detection.py
View File

@ -52,7 +52,12 @@ class LocalObjectDetector(ObjectDetector):
else: else:
self.input_transform = None 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): def detect(self, tensor_input, threshold=0.4):
detections = [] detections = []