aidas-79 commited on
Commit
24c0734
·
verified ·
1 Parent(s): 82b673e

Update video_detection.py

Browse files
Files changed (1) hide show
  1. 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
- #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,9 +98,9 @@ def video_detection(cap):
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_{uuid.uuid4()}.mp4"
103
- segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height)) # type: ignore
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