Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-04 23:14:12 +02:00
Code Issues Projects Releases Wiki Activity

Map verified objects to their sub label directly (#20413)

Browse Source 
* Map verified objects to their sub label directly

* Simplify access

* Cleanup

* Add protection for mismatched object and index

* Keep track of verified objects separately
This commit is contained in:
Nicolas Mowen
2025-10-10 07:07:00 -06:00
committed by GitHub
parent 24a1874225
commit b1a5896b53
3 changed files with 23 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -255,19 +255,24 @@ def run_analysis(
}
objects = []
verified_objects = []
named_objects = []
for label in set(final_data["data"]["objects"] + final_data["data"]["sub_labels"]):
objects_list = final_data["data"]["objects"]
sub_labels_list = final_data["data"]["sub_labels"]
for label in objects_list:
if "-verified" in label:
continue
if label in labelmap_objects:
elif label in labelmap_objects:
objects.append(label.replace("_", " ").title())
else:
verified_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["recognized_objects"] = verified_objects
analytics_data["recognized_objects"] = named_objects
metadata = genai_client.generate_review_description(
analytics_data,