Spaces:
Runtime error
Runtime error
Update video_detection.py
Browse files- video_detection.py +5 -5
video_detection.py
CHANGED
@@ -46,8 +46,8 @@ def video_detection(cap):
|
|
46 |
|
47 |
n_frames = 0
|
48 |
|
49 |
-
name = f"output_{uuid.uuid4()}.mp4"
|
50 |
-
|
51 |
segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height)) # type: ignore
|
52 |
batch = []
|
53 |
while iterating:
|
@@ -98,9 +98,9 @@ def video_detection(cap):
|
|
98 |
segment_file.release()
|
99 |
n_frames = 0
|
100 |
yield fame, name
|
101 |
-
|
102 |
-
|
103 |
-
|
104 |
else:
|
105 |
yield fame, None
|
106 |
|
|
|
46 |
|
47 |
n_frames = 0
|
48 |
|
49 |
+
#name = f"output_{uuid.uuid4()}.mp4"
|
50 |
+
name = f"output_{n_chunks}.mp4" #if stream_as_mp4 else '.ts'}
|
51 |
segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height)) # type: ignore
|
52 |
batch = []
|
53 |
while iterating:
|
|
|
98 |
segment_file.release()
|
99 |
n_frames = 0
|
100 |
yield fame, name
|
101 |
+
#name = f"output_{n_chunks}{'.mp4' if stream_as_mp4 else '.ts'}"
|
102 |
+
name = f"output_{n_chunks}.mp4"
|
103 |
+
segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height)) # type: ignore
|
104 |
else:
|
105 |
yield fame, None
|
106 |
|