update export video

LHM/utils/ffmpeg_utils.py

@@ -36,7 +36,7 @@ def encodeffmpeg(inputs, frame_rate, output, format="png"):
     os.system(cmd)
     print("video done!")
 
-def images_to_video(images, output_path, fps, gradio_codec: bool, verbose=False, bitrate="5M"):
+def images_to_video(images, output_path, fps, gradio_codec: bool, verbose=False, bitrate="2M"):
     os.makedirs(os.path.dirname(output_path), exist_ok=True)
     frames = []
     for i in range(images.shape[0]):
@@ -46,9 +46,13 @@ def images_to_video(images, output_path, fps, gradio_codec: bool, verbose=False,
                 f"Frame shape mismatch: {frame.shape} vs {images.shape}"
             assert frame.min() >= 0 and frame.max() <= 255, \
                 f"Frame value out of range: {frame.min()} ~ {frame.max()}"
+            
         else:
             frame = images[i]
-        frames.append(frame)
+        width, height = frame.shape[1], frame.shape[0]
+        # reshape to limit the export time
+        if width > 1200 or height > 1200 or images.shape[0] > 400:
+            frames.append(cv2.resize(frame, (width // 2, height // 2)))
     
     frames = np.stack(frames)
     iio.imwrite(output_path,frames,fps=fps,codec="libx264",pixelformat="yuv420p",bitrate=bitrate,macro_block_size=16)