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

Update DeepStack plugin to recognize 'truck' as 'car' for label indexing

Browse Source
This commit is contained in:
Sergey Krashevich 2023-04-23 17:21:45 +03:00
parent 445315e9e3
commit f02a9d6238
No known key found for this signature in database
GPG Key ID: 625171324E7D3856
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
frigate/detectors/plugins/deepstack.py
View File

@ -55,10 +55,12 @@ class DeepStack(DetectionApi):
return labels
def get_label_index(self, label_value):
if label_value.lower() == 'truck':
label_value = 'car'
for index, value in self.labels.items():
if value == label_value:
if value == label_value.lower():
return index
return None
return -1
def detect_raw(self, tensor_input):
image_data = np.squeeze(tensor_input).astype(np.uint8)
@ -74,8 +76,11 @@ class DeepStack(DetectionApi):
for i, detection in enumerate(response_json["predictions"]):
if detection["confidence"] < 0.4:
break
label = self.get_label_index(detection["label"])
if label < 0:
break
detections[i] = [
int(self.get_label_index(detection["label"])),
label,
float(detection["confidence"]),
detection["y_min"] / self.h,
detection["x_min"] / self.w,