|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import os |
|
import numpy as np |
|
import imageio |
|
|
|
|
|
def images_to_video(images, output_path, fps, gradio_codec: bool, verbose=False): |
|
|
|
os.makedirs(os.path.dirname(output_path), exist_ok=True) |
|
frames = [] |
|
for i in range(images.shape[0]): |
|
frame = (images[i].permute(1, 2, 0).cpu().numpy() * 255).astype(np.uint8) |
|
assert frame.shape[0] == images.shape[2] and frame.shape[1] == images.shape[3], \ |
|
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()}" |
|
frames.append(frame) |
|
frames = np.stack(frames) |
|
if gradio_codec: |
|
imageio.mimwrite(output_path, frames, fps=fps, quality=10) |
|
else: |
|
imageio.mimwrite(output_path, frames, fps=fps, codec='mpeg4', quality=10) |
|
if verbose: |
|
print(f"Using gradio codec option {gradio_codec}") |
|
print(f"Saved video to {output_path}") |
|
|