95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Convert MP4 videos to optimized GIFs for GitHub README display.
|
|
This script creates smaller, optimized GIFs suitable for web display.
|
|
"""
|
|
import os
|
|
import sys
|
|
import imageio
|
|
from pathlib import Path
|
|
|
|
def convert_mp4_to_gif(mp4_path, gif_path, max_frames=150, fps=10, scale=0.5):
|
|
"""
|
|
Convert MP4 to optimized GIF.
|
|
|
|
Args:
|
|
mp4_path: Path to input MP4 file
|
|
gif_path: Path to output GIF file
|
|
max_frames: Maximum number of frames to include (for file size)
|
|
fps: Frames per second for output GIF
|
|
scale: Scale factor (0.5 = half size)
|
|
"""
|
|
print(f"Converting {mp4_path} to {gif_path}...")
|
|
|
|
try:
|
|
# Read video
|
|
reader = imageio.get_reader(mp4_path)
|
|
fps_original = reader.get_meta_data()['fps']
|
|
|
|
# Calculate frame skip to get desired fps
|
|
frame_skip = max(1, int(fps_original / fps))
|
|
|
|
frames = []
|
|
frame_count = 0
|
|
|
|
for i, frame in enumerate(reader):
|
|
if i % frame_skip == 0 and frame_count < max_frames:
|
|
# Resize frame
|
|
import numpy as np
|
|
from PIL import Image
|
|
img = Image.fromarray(frame)
|
|
new_size = (int(img.width * scale), int(img.height * scale))
|
|
img_resized = img.resize(new_size, Image.Resampling.LANCZOS)
|
|
frames.append(np.array(img_resized))
|
|
frame_count += 1
|
|
|
|
reader.close()
|
|
|
|
# Write GIF
|
|
imageio.mimsave(gif_path, frames, fps=fps, loop=0)
|
|
print(f" ✓ Created {gif_path} ({frame_count} frames, {fps} fps)")
|
|
|
|
# Get file sizes
|
|
mp4_size = os.path.getsize(mp4_path) / (1024 * 1024) # MB
|
|
gif_size = os.path.getsize(gif_path) / (1024 * 1024) # MB
|
|
print(f" Size: {mp4_size:.1f} MB → {gif_size:.1f} MB")
|
|
|
|
except Exception as e:
|
|
print(f" ✗ Error: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
def main():
|
|
assets_dir = Path(__file__).parent.parent / "assets"
|
|
|
|
videos = [
|
|
"droid_result.mp4",
|
|
"vipe_result.mp4",
|
|
"handedness1.mp4",
|
|
"handedness2.mp4",
|
|
"handedness3.mp4",
|
|
"handedness4.mp4",
|
|
]
|
|
|
|
print("Converting videos to GIFs for GitHub README...")
|
|
print("=" * 60)
|
|
|
|
for video in videos:
|
|
mp4_path = assets_dir / video
|
|
gif_path = assets_dir / video.replace(".mp4", ".gif")
|
|
|
|
if not mp4_path.exists():
|
|
print(f"⚠ Skipping {video} (not found)")
|
|
continue
|
|
|
|
convert_mp4_to_gif(mp4_path, gif_path, max_frames=1000, fps=20, scale=0.6)
|
|
print()
|
|
|
|
print("=" * 60)
|
|
print("Done! Update README.md to use .gif files instead of .mp4")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|