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

Rename to detect_api

Browse Source 
Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com>
This commit is contained in:
Nate Meyer 2022-08-15 21:27:45 -04:00 committed by Nate Meyer
parent 2c6e96d9f7
commit dc0b2d62ae
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/object_detection.py
View File

@ -41,12 +41,12 @@ class LocalObjectDetector(ObjectDetector):
self.labels = load_labels(labels)
if det_type == DetectorTypeEnum.edgetpu:
self.detectApi = EdgeTpuTfl(det_device=det_device, model_path=model_path)
self.detect_api = EdgeTpuTfl(det_device=det_device, model_path=model_path)
else:
logger.warning(
"CPU detectors are not recommended and should only be used for testing or for trial purposes."
)
self.detectApi = CpuTfl(model_path=model_path, num_threads=num_threads)
self.detect_api = CpuTfl(model_path=model_path, num_threads=num_threads)
def detect(self, tensor_input, threshold=0.4):
detections = []
@ -63,7 +63,7 @@ class LocalObjectDetector(ObjectDetector):
return detections
def detect_raw(self, tensor_input):
return self.detectApi.detect_raw(tensor_input=tensor_input)
return self.detect_api.detect_raw(tensor_input=tensor_input)
def run_detector(