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

이 스크립트는 LoRA, 체크포인트 등에 대한 주요 데이터셋 도우미입니다!

YouTube URL(다른 여러 사이트도 지원!) 또는 로컬 비디오 파일을 제공하면, 자동으로 최고 품질의 스크린샷을 추출하고 미적 기준에 따라 자동 필터링하는 등의 기능을 제공합니다!

자세한 내용: https://github.com/EnragedAntelope/youtube-screenshot-extractor

기능:

• yt-dlp를 사용하여 YouTube 및 기타 비디오 다운로드

• 로컬 비디오 파일 처리

• 여러 프레임 추출 방법:

  • 간격 기반 추출

  • 모든 프레임 추출

  • 키프레임 추출

  • 장면 감지

• 프레임 품질 평가

• 흐림 감지

• 자동 검은 바 제거

• 기본 워터마크 감지

• 병렬 처리로 빠른 실행

• GPU 가속(사용 가능한 경우)

• 중단된 추출 재개

• 썸네일 몬타주 생성

• 사용자 정의 출력 옵션(JPG 또는 PNG)

• 상세 로깅 및 드라이런 옵션

• 구성 파일에서 설정 로드

• 후처리 필터:

  • Gradfun(컬러 밴딩 감소, 덜 공격적)

  • Deblock(압축 아티팩트 감소)

  • Deband(컬러 밴딩 감소, 더 공격적)