CoolFace
Apppublic

xSICHx/human2simpson_x_horse2zebra_CycleGAN

sourceHugging Facemitupdated 5mo agoView on Hugging Face
0likes
App README

Задача: Человек ↔ Симпсон

Перенос стиля между фотографиями лиц людей (домен A) и кадрами персонажей «Симпсонов» (домен B).

Также дополнительно предоставлена возможность преобразования Лошадь ↔ Зебра

Структура репозитория

.
├── Dockerfile              
├── requirements.txt       
├── README.md
│
├── src/                    — код инференса
│   ├── streamlit_app.py    — UI
│   ├── model.py            — архитектура CycleGAN + загрузчики чекпоинтов
│   ├── inference.py        — препроцессинг, постпроцессинг, CycleGANInferencer
│   └── configs.py          — конфиги моделей (MODEL_CONFIGS)
│
├── checkpoints/            — веса моделей
│   ├── cycle_gan_simpsons.pt
│   └── cycle_gan_zebras.pt
│
├── photos/                 — примеры для каждого направления перевода
│   ├── real2simpson/       — A→B для симпсонов
│   ├── simpson2real/       — B→A для симпсонов
│   ├── horse2zebra/        — A→B для зебр
│   └── zebra2horse/        — B→A для зебр
│
├── training/               — код обучения моделей
│   ├── cyclegan_human2simpson_train.ipynb
│   └── cyclegan_horse2zebra_train.ipynb
│
└── person2simpson_data.zip — архив датасета для собственной задачи (человек ↔ Симпсон)