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:
Simple startup .bat/scripts for Windows, Mac, and Linux!
Easy to use GUI front-end!
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)
Description
Massive overhaul with tons of new features, including ability to process local video, set frame extraction rules (based on scene, interval, keyframe, etc.), blur threshrold improvement, ability to detect watermarks, parallel processing, and much more!