Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
ae211e5
1
Parent(s):
a8dfdef
fix bug
Browse files
LHM/utils/ffmpeg_utils.py
CHANGED
@@ -54,7 +54,7 @@ def images_to_video(images, output_path, fps, gradio_codec: bool, verbose=False,
|
|
54 |
# frames.append(cv2.resize(frame, (width // 2, height // 2)))
|
55 |
|
56 |
# limit the frames directly @NOTE huggingface only!
|
57 |
-
|
58 |
|
59 |
frames = np.stack(frames)
|
60 |
iio.imwrite(output_path,frames,fps=fps,codec="libx264",pixelformat="yuv420p",bitrate=bitrate,macro_block_size=16)
|
|
|
54 |
# frames.append(cv2.resize(frame, (width // 2, height // 2)))
|
55 |
|
56 |
# limit the frames directly @NOTE huggingface only!
|
57 |
+
frames = frames[:300]
|
58 |
|
59 |
frames = np.stack(frames)
|
60 |
iio.imwrite(output_path,frames,fps=fps,codec="libx264",pixelformat="yuv420p",bitrate=bitrate,macro_block_size=16)
|