administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-12 01:57:36 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Keep track of verified objects separately

Browse Source
This commit is contained in:
Nicolas Mowen 2025-10-10 06:52:26 -06:00
parent 7dd0cadaae
commit 74f7251d6b
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
frigate/data_processing/post/review_descriptions.py
View File

@ -259,20 +259,18 @@ def run_analysis(
objects_list = final_data["data"]["objects"] objects_list = final_data["data"]["objects"]
sub_labels_list = final_data["data"]["sub_labels"] sub_labels_list = final_data["data"]["sub_labels"]
verified_index = 0
for label in objects_list: for label in objects_list:
if "-verified" in label: if "-verified" in label:
try: continue
named_objects.append(
f"{sub_labels_list[verified_index].replace('_', ' ').title()} ({label.replace('-verified', '')})"
)
verified_index += 1
except IndexError:
continue
elif label in labelmap_objects: elif label in labelmap_objects:
objects.append(label.replace("_", " ").title()) objects.append(label.replace("_", " ").title())
for i, verified_label in enumerate(final_data["data"]["verified_objects"]):
named_objects.append(
f"{sub_labels_list[i].replace('_', ' ').title()} ({verified_label.replace('-verified', '')})"
)
analytics_data["objects"] = objects analytics_data["objects"] = objects
analytics_data["recognized_objects"] = named_objects analytics_data["recognized_objects"] = named_objects

3
frigate/review/maintainer.py
View File

@ -147,6 +147,9 @@ class PendingReviewSegment:
ReviewSegment.data.name: { ReviewSegment.data.name: {
"detections": list(set(self.detections.keys())), "detections": list(set(self.detections.keys())),
"objects": list(set(self.detections.values())), "objects": list(set(self.detections.values())),
"verified_objects": [
o for o in self.detections.values() if "-verified" in o
],
"sub_labels": list(self.sub_labels.values()), "sub_labels": list(self.sub_labels.values()),
"zones": self.zones, "zones": self.zones,
"audio": list(self.audio), "audio": list(self.audio),