README.md

metadata

title: Face Dataset Builder
emoji: 🎭
colorFrom: indigo
colorTo: purple
sdk: gradio
app_file: app.py
pinned: false

Сборщик датасета лиц из видео под LoRA

Загрузи видео — получишь разнообразный чистый набор лиц для обучения LoRA.

Что делает: режет видео → детектит лицо (InsightFace) → отсекает смазанное → кластеризует по ракурсам (KMeans по эмбеддингам, чтобы не было 30 одинаковых фронталок) → кропает и выравнивает → отдаёт галерею + zip.

Локально (есть GPU): python app.py, доступен флаг апскейла мелких лиц через Real-ESRGAN. На облачном CPU апскейл отключён — слишком медленно.

CLI без морды:

python build_face_dataset.py input.mp4 --num 30 --target 20
python build_face_dataset.py input.mp4 --num 30 --upscale --full-quality   # локально на GPU