| 25 | |
| 26 | |
| 27 | def load_images_from_folder(folder, width, height): |
| 28 | images = [] |
| 29 | files = os.listdir(folder) |
| 30 | png_files = [f for f in files if f.endswith('.png')] |
| 31 | png_files.sort(key=lambda x: int(x.split('_')[1].split('.')[0])) |
| 32 | for filename in png_files: |
| 33 | img = Image.open(os.path.join(folder, filename)).convert('RGB') |
| 34 | img = img.resize((width, height)) |
| 35 | images.append(img) |
| 36 | |
| 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] |