Update periodic_detection_function.py

periodic_detection_function.py

@@ -189,8 +189,8 @@ def run_periodic_detection(video_path, trajectory_path, output_video_path=None,
 
     ### Task 1
     period_num = len([x for x in seg_labels.values() if len(x)>0.5*win])
-    print("period_num: {}".format(period_num))
-    print("seg_labels_index: {}".format(seg_labels.keys()))
+    #print("period_num: {}".format(period_num))
+    #print("seg_labels_index: {}".format(seg_labels.keys()))
     if period_num>0:
         period_boundaries = {}
         for p_id, (k,v) in enumerate(seg_labels.items()):
@@ -293,7 +293,7 @@ def run_periodic_detection(video_path, trajectory_path, output_video_path=None,
             print("Error: Could not open video writer with any compatible codec.")
         
         i, j = 0, 0
-        for k in tqdm(list(seg_labels.keys())):
+        for idx, k in enumerate(tqdm(list(seg_labels.keys()))):
             if not seg_labels[k]:  # Skip empty segments
                 continue
                 
@@ -301,6 +301,21 @@ def run_periodic_detection(video_path, trajectory_path, output_video_path=None,
             frame_ids = list(seg_labels[k].keys())
             j += len(seg_labels[k])
             
+            # Use boundaries from JSON (period_boundaries) if available, otherwise fallback or match
+            start_frame_text = "????"
+            end_frame_text = "????"
+            # period_boundaries handles both dict (from detection) and list (fallback)
+            # Keys or indices matching the segment order
+            try:
+                # period_boundaries might be dict or list. 
+                # If dict, keys usually match iteration order of seg_labels.items() if consistent
+                # If list, idx matches sequence
+                boundary = period_boundaries[idx]
+                start_frame_text = f"{boundary[0]:04d}"
+                end_frame_text = f"{boundary[1]:04d}"
+            except:
+                pass
+
             cv2.putText(progress_bar, f'Period {k+1}', (5, height*k+30), 
                        cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 0), 2)
             
@@ -324,15 +339,27 @@ def run_periodic_detection(video_path, trajectory_path, output_video_path=None,
                     print(f"Error in video generation: {str(e)}")
                     continue
             
-            cv2.putText(progress_bar, f'Frame: {(i+1):04d}-{j:04d}', (5, height*k+52), 
+            cv2.putText(progress_bar, f'Frame: {start_frame_text}-{end_frame_text}', (5, height*k+52), 
                        cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 0), 2)
             i += len(seg_labels[k])
-            
-            try:
-                out.write(frame)
-            except:
-                pass
         
+        # Add freeze frame at the end to show the final state
+        try:
+            # Reconstruct the final frame with the updated progress_bar (which has the last period's boundaries text)
+            # frame structure: Top part (Video + ProgressBar), Bottom part (Anchor)
+            # We assume 'frame' holds the last written frame. We extract the video component (Top-Left).
+            # Video part is [:H, :W]
+            if frame is not None:
+                last_video_part = frame[:H, :W, :]
+                top_part = np.concatenate([last_video_part, (progress_bar*255).astype(np.uint8)[:,:,::-1]], axis=1)
+                final_frame = np.concatenate([top_part, anchor], axis=0)
+                
+                for _ in range(90): # 3 seconds pause
+                    out.write(final_frame)
+        except Exception as e:
+            print(f"Error creating freeze frame: {e}")
+            pass
+
         # Release resources
         cap.release()
         out.release()