Dataset helper for loras or checkpoints! Auto capture and filter high quality screengrabs!

This script is a major dataset helper for loras, checkpoints, etc.!

Provide any youtube (many other sites also work!) URL OR local video file, and it will automatically extract the highest-available-quality screenshots, auto filter for aesthetics, and more!

Full details: https://github.com/EnragedAntelope/youtube-screenshot-extractor

Features:

• Download YouTube and other videos using yt-dlp

• Process local video files

• Multiple frame extraction methods:

  • Interval-based extraction

  • All frames extraction

  • Keyframe extraction

  • Scene detection

• Quality assessment of frames

• Blur detection

• Automatic removal of black bars

• Basic watermark detection

• Parallel processing for faster execution

• GPU acceleration (if available)

• Resume interrupted extractions

• Generate thumbnail montages

• Customizable output options (JPG or PNG)

• Detailed logging and dry-run option

• Load settings from a configuration file

• Post-processing filters:

  • Gradfun (reduce color banding, less aggressive)

  • Deblock (reduce compression artifacts)

  • Deband (reduce color banding, more aggressive)