Spaces:
Sleeping
Sleeping
Sean Carnahan commited on
Commit ·
460d366
1
Parent(s): 146de98
Use mp4v codec for VideoWriter
Browse files
app.py
CHANGED
|
@@ -221,7 +221,7 @@ def process_video_movenet(video_path, model_variant='lightning', pose_type='fron
|
|
| 221 |
output_path = os.path.join(app.config['UPLOAD_FOLDER'], output_filename)
|
| 222 |
print(f"Output path: {output_path}")
|
| 223 |
|
| 224 |
-
fourcc = cv2.VideoWriter_fourcc(*'
|
| 225 |
out = cv2.VideoWriter(output_path, fourcc, fps, (total_width, height))
|
| 226 |
if not out.isOpened():
|
| 227 |
raise ValueError(f"Failed to create output video writer at {output_path}")
|
|
@@ -383,7 +383,7 @@ def process_video_mediapipe(video_path):
|
|
| 383 |
print(f"Processing video with MediaPipe: {width}x{height} @ {fps}fps")
|
| 384 |
output_filename = f'output_mediapipe.mp4'
|
| 385 |
output_path = os.path.join(app.config['UPLOAD_FOLDER'], output_filename)
|
| 386 |
-
fourcc = cv2.VideoWriter_fourcc(*'
|
| 387 |
out = cv2.VideoWriter(output_path, fourcc, fps, (total_width, height))
|
| 388 |
if not out.isOpened():
|
| 389 |
raise ValueError(f"Failed to create output video writer at {output_path}")
|
|
|
|
| 221 |
output_path = os.path.join(app.config['UPLOAD_FOLDER'], output_filename)
|
| 222 |
print(f"Output path: {output_path}")
|
| 223 |
|
| 224 |
+
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
|
| 225 |
out = cv2.VideoWriter(output_path, fourcc, fps, (total_width, height))
|
| 226 |
if not out.isOpened():
|
| 227 |
raise ValueError(f"Failed to create output video writer at {output_path}")
|
|
|
|
| 383 |
print(f"Processing video with MediaPipe: {width}x{height} @ {fps}fps")
|
| 384 |
output_filename = f'output_mediapipe.mp4'
|
| 385 |
output_path = os.path.join(app.config['UPLOAD_FOLDER'], output_filename)
|
| 386 |
+
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
|
| 387 |
out = cv2.VideoWriter(output_path, fourcc, fps, (total_width, height))
|
| 388 |
if not out.isOpened():
|
| 389 |
raise ValueError(f"Failed to create output video writer at {output_path}")
|