CatVTON — виртуальная примерка одежды

★ 1.9K

CatVTON — это открытая диффузионная модель виртуальной примерки одежды (принята на ICLR 2025): простая и лёгкая, 899 млн параметров, из них обучаемых 49,57 млн. Берут, когда нужно показать, как одежда сидит на конкретном человеке без фотосессии — для карточек интернет-магазина, лукбуков и примерочных; когда примерку надо запускать у себя, не отправляя фото в облако; когда видеопамяти мало. Как использовать: conda-окружение с Python 3.9 и pip install -r requirements.txt, Gradio-приложение app.py, воркфлоу для ComfyUI, скрипт inference.py для датасетов VITON-HD и DressCode; веса скачиваются с Hugging Face автоматически. В режиме bf16 генерация 1024×768 требует около 8 ГБ VRAM. Есть CatVTON-FLUX — облегчённая LoRA на 37,4 млн параметров для FLUX.1-Fill-dev, и версия без маски (Mask-free). Для примерки на видео авторы предлагают отдельную модель CatV2TON. Направление — надеть одежду на фото человека, а не создать одежду с нуля. Нужна видеокарта с CUDA. Лицензию репозитория автоматически определить не удалось — перед коммерческим использованием проверьте её условия. Применять только с согласия людей, чьи фото обрабатываются.