Update video_detection.py

video_detection.py

@@ -1,5 +1,9 @@
 from ultralytics import YOLO
 import cv2
+from PIL import Image
+import time
+import numpy as np
+import uuid
 
 model = YOLO("model/yolo11n_6-2-25.pt")
 
@@ -25,15 +29,38 @@ def draw_boxes(frame, results):
             cv2.putText(frame, cls, org, font, fontScale, color, thickness)
 
     return frame
+    
+def video_detection(cap):
+    frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
+
+    cap = cv2.VideoCapture(video)
 
+    video_codec = cv2.VideoWriter_fourcc(*"mp4v") # type: ignore
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
 
+    desired_fps = fps // SUBSAMPLE
+    width  = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)) // 2
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) // 2
+
+    iterating, frame = cap.read()
+
+    n_frames = 0
+
+    name = f"output_{uuid.uuid4()}.mp4"
+    segment_file = cv2.VideoWriter(name, video_codec, desired_fps, (width, height)) # type: ignore
+    batch = []
+
+"""
+#@spaces.GPU
 def video_detection(cap):
     frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     fps = int(cap.get(cv2.CAP_PROP_FPS))
 
     out = cv2.VideoWriter('output_video.mp4', cv2.VideoWriter_fourcc(*'h264'), fps, (frame_width, frame_height))
-
+    
     count = 0
     while cap.isOpened():
         success, frame = cap.read()
@@ -57,3 +84,4 @@ def video_detection(cap):
     cv2.destroyAllWindows()
 
     yield None, 'output_video.mp4'
+"""