nosopirka/cyclegan-day-night

CycleGAN Day ↔ Night

Модель для преобразования дневных городских фото в ночные и обратно.

Архитектура

• Генераторы: ResNet-9 blocks (256×256)
• Дискриминаторы: PatchGAN + Global Average Pooling
• Loss: LSGAN + Cycle Consistency (λ=10)

Нормализация (домен A — день)

• mean: [0.49986013770103455, 0.5164332985877991, 0.4983072280883789]
• std: [0.2775309681892395, 0.26964136958122253, 0.30012083053588867]

Нормализация (домен B — ночь)

• mean: [0.2502242922782898, 0.2704642415046692, 0.31721022725105286]
• std: [0.24460846185684204, 0.20378708839416504, 0.22527939081192017]

Использование

from huggingface_hub import hf_hub_download
import torch

gen = Generator()
gen.load_state_dict(torch.load(
    hf_hub_download("nosopirka/cyclegan-day-night", "gen_a_to_b.pth"), map_location="cpu"
))
gen.eval()