Views
No views yet
├── checkpoints/
│ └── cyclegan_cctv/
│ └── G_cctv.pth # Generator model (Clean → Noisy)
├── outputs/
│ └── cyclegan_cctv/ # Training outputs & visualizations
├── coco_clean/ # Original COCO images (1000 samples)
├── coco_noisy/ # CycleGAN-generated noisy images (1000 samples)
└── README.md1from huggingface_hub import hf_hub_download
2import torch
3
4# Download model
5model_path = hf_hub_download(
6 repo_id="dtann08/cyclegan-cctv-noise-generator",
7 filename="checkpoints/cyclegan_cctv/G_cctv.pth"
8)
9
10# Load model (define Generator class first)
11G_AB = Generator(input_nc=3, output_nc=3, n_residual_blocks=6)
12G_AB.load_state_dict(torch.load(model_path, map_location='cpu'))
13G_AB.eval()1from huggingface_hub import snapshot_download
2
3# Download specific folders
4snapshot_download(
5 repo_id="dtann08/cyclegan-cctv-noise-generator",
6 allow_patterns=["coco_clean/*", "coco_noisy/*"],
7 local_dir="./data"
8)