diff --git a/frigate/data_processing/post/review_descriptions.py b/frigate/data_processing/post/review_descriptions.py index f924f1a89..69110564c 100644 --- a/frigate/data_processing/post/review_descriptions.py +++ b/frigate/data_processing/post/review_descriptions.py @@ -259,20 +259,18 @@ def run_analysis( 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: - try: - named_objects.append( - f"{sub_labels_list[verified_index].replace('_', ' ').title()} ({label.replace('-verified', '')})" - ) - verified_index += 1 - except IndexError: - continue + continue elif label in labelmap_objects: 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"] = named_objects diff --git a/frigate/review/maintainer.py b/frigate/review/maintainer.py index 202b9b701..40d7b64cf 100644 --- a/frigate/review/maintainer.py +++ b/frigate/review/maintainer.py @@ -147,6 +147,9 @@ class PendingReviewSegment: ReviewSegment.data.name: { "detections": list(set(self.detections.keys())), "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()), "zones": self.zones, "audio": list(self.audio),