changing the variables of function summary_from_frame_list relevant t…

…o the change of input dictionary
odigous-labs · Mar 2, 2019 · acc8f0b · acc8f0b
1 parent f3b4ca5
commit acc8f0b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/summary_from_frame_list.py b/src/summary_from_frame_list.py
@@ -6,9 +6,11 @@ def summary_from_frame_list(frame_list_with_objects, threshold_of_objects_in_fra
         obj_list = frame_list_with_objects[list(frame_list_with_objects.keys())[i]]
 
         for j in range(threshold_of_objects_in_frame):
-            if (obj_list[j][0] not in object_array):
-                object_array.append(obj_list[j][0])
-                if (i not in summary_frames):
-                    summary_frames.append(i)
+            if (obj_list[j] not in object_array):
+                object_array.append(obj_list[j])
+
+                frame_name = list(frame_list_with_objects.keys())[i]
+                if (frame_name not in summary_frames):
+                    summary_frames.append(frame_name)
 
     return summary_frames