xSICHx/human2simpson_x_horse2zebra_CycleGAN
0
Задача: Человек ↔ Симпсон
Перенос стиля между фотографиями лиц людей (домен 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 — архив датасета для собственной задачи (человек ↔ Симпсон)
