letsolum/usda-win2sum

CycleGAN Img2Img

Demo для лабораторной по CycleGAN (unpaired image-to-image translation).

Что внутри

• —summer2winter_yosemite — классический датасет Berkeley, модель преобразует летние пейзажи в зимние и наоборот.
• —selfie2anime — селфи ↔ аниме-портреты, обучено на датасете arnaud58/selfie2anime (~3400 картинок на домен).

Архитектура

Классический CycleGAN (Zhu et al., 2017):

• —Два ResNet-генератора с 9 residual-блоками
• —Два PatchGAN-дискриминатора (70×70)
• —LSGAN adversarial + L1 cycle (λ=10) + L1 identity (λ=5)
• —256×256, 80 epochs, linear LR decay с эпохи 40

Как использовать

1. 1.Выберите модель в сайдбаре
2. 2.Выберите направление (A→B или B→A)
3. 3.Загрузите картинку
4. 4.Получите результат