From 0f8c3407e75225ef6cea541e6d841858047224cf Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Fri, 10 Oct 2025 06:30:02 -0600 Subject: [PATCH] Simplify access --- .../post/review_descriptions.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/frigate/data_processing/post/review_descriptions.py b/frigate/data_processing/post/review_descriptions.py index ba34d9a20..9a27b727c 100644 --- a/frigate/data_processing/post/review_descriptions.py +++ b/frigate/data_processing/post/review_descriptions.py @@ -255,20 +255,21 @@ 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"] + verified_index = 0 + + for label in objects_list: if "-verified" in label: - verified_objects.append(label) - continue - - if label in labelmap_objects: + named_objects.append( + f"{sub_labels_list[verified_index].replace('_', ' ').title()} ({label.replace('-verified', '')})" + ) + verified_index += 1 + elif label in labelmap_objects: objects.append(label.replace("_", " ").title()) - for label in verified_objects: - named_objects.append(f'{label.replace("_", " ").title()} ({label.replace("-verified", "")})') - analytics_data["objects"] = objects analytics_data["recognized_objects"] = named_objects