(frames, save_path, fps, quality=9, ffmpeg_params=None)
| 136 | |
| 137 | |
| 138 | def save_video(frames, save_path, fps, quality=9, ffmpeg_params=None): |
| 139 | writer = imageio.get_writer(save_path, fps=fps, quality=quality, ffmpeg_params=ffmpeg_params) |
| 140 | for frame in tqdm(frames, desc="Saving video"): |
| 141 | frame = np.array(frame) |
| 142 | writer.append_data(frame) |
| 143 | writer.close() |
| 144 | |
| 145 | def save_frames(frames, save_path): |
| 146 | os.makedirs(save_path, exist_ok=True) |