(frames, output_path)
| 37 | return images |
| 38 | |
| 39 | def save_frames_as_png(frames, output_path): |
| 40 | pil_frames = [Image.fromarray(frame) if isinstance(frame, np.ndarray) else frame for frame in frames] |
| 41 | num_frames = len(pil_frames) |
| 42 | for i in range(num_frames): |
| 43 | pil_frame = pil_frames[i] |
| 44 | save_path = os.path.join(output_path, f'frame_{i}.png') |
| 45 | pil_frame.save(save_path) |
| 46 | |
| 47 | def save_frames_as_mp4(frames, output_mp4_path, fps): |
| 48 | print("Starting saving the frames as mp4") |