Update video_detection.py

video_detection.py

@@ -47,6 +47,7 @@ def video_detection(cap):
     n_frames = 0
 
     name = f"output_{uuid.uuid4()}.mp4"
+    #name = f"output_{n_chunks}{'.mp4' if stream_as_mp4 else '.ts'}"
     segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height)) # type: ignore
     batch = []
     while iterating:
@@ -79,6 +80,9 @@ def video_detection(cap):
             frame = frame[:, :, ::-1].copy()
             segment_file.write(frame)
         """
+        frame = draw_boxes(frame, results)
+        segment_file.write(frame)
+        """
         for i, r in enumerate(results):
             # Plot results image
             im_bgr = r.plot()  # BGR-order numpy array
@@ -87,15 +91,18 @@ def video_detection(cap):
             # Convert RGB to BGR
             frame = frame[:, :, ::-1].copy()
             segment_file.write(frame)
+        """
             
-        #if n_frames == 3 * fps:
-        #    n_chunks += 1
-        #    segment_file.release()
-        #    n_frames = 0
-        yield  frame, None
-        #name = f"output_{n_chunks}{'.mp4' if stream_as_mp4 else '.ts'}"
-        #name = f"output_{uuid.uuid4()}.mp4"
-        #segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height)) # type: ignore
+        if n_frames == 16:
+             n_chunks += 1
+             segment_file.release()
+             n_frames = 0
+             yield  fame, name
+            #name = f"output_{n_chunks}{'.mp4' if stream_as_mp4 else '.ts'}"
+            name = f"output_{uuid.uuid4()}.mp4"
+            segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height)) # type: ignore
+        else:
+            yield  fame, None
 
         #batch = []
         #segment_file.release()