Spaces:
Running on Zero
Running on Zero
pixel-format-fix (#3)
Browse files- update pixel format (0b97fc423c2a2b5ba36d8fec8c37b8e6fcff1a91)
- demo_utils.py +2 -2
demo_utils.py
CHANGED
|
@@ -39,7 +39,7 @@ def image_to_video(image_path, output_path, fps):
|
|
| 39 |
|
| 40 |
img_array = np.array(img)
|
| 41 |
|
| 42 |
-
with imageio.get_writer(output_path, fps=fps, format='FFMPEG', codec='libx264', pixelformat='
|
| 43 |
writer.append_data(img_array)
|
| 44 |
|
| 45 |
return {
|
|
@@ -582,6 +582,6 @@ def process_video(video_path, setup, gazing_ratio=0.75, task_loss_requirement=0.
|
|
| 582 |
|
| 583 |
|
| 584 |
def save_video(frames, output_path, fps):
|
| 585 |
-
with imageio.get_writer(output_path, fps=fps, format='FFMPEG', codec='libx264', pixelformat='
|
| 586 |
for frame in frames:
|
| 587 |
writer.append_data(frame)
|
|
|
|
| 39 |
|
| 40 |
img_array = np.array(img)
|
| 41 |
|
| 42 |
+
with imageio.get_writer(output_path, fps=fps, format='FFMPEG', codec='libx264', pixelformat='yuv444p', macro_block_size=1) as writer:
|
| 43 |
writer.append_data(img_array)
|
| 44 |
|
| 45 |
return {
|
|
|
|
| 582 |
|
| 583 |
|
| 584 |
def save_video(frames, output_path, fps):
|
| 585 |
+
with imageio.get_writer(output_path, fps=fps, format='FFMPEG', codec='libx264', pixelformat='yuv444p', macro_block_size=1) as writer:
|
| 586 |
for frame in frames:
|
| 587 |
writer.append_data(frame)
|