{"id":18071893,"url":"https://github.com/innerlee/ganbase","last_synced_at":"2025-04-05T17:41:43.778Z","repository":{"id":57433020,"uuid":"120847644","full_name":"innerlee/ganbase","owner":"innerlee","description":"tools for GAN research","archived":false,"fork":false,"pushed_at":"2019-06-20T09:22:44.000Z","size":10589,"stargazers_count":1,"open_issues_count":1,"forks_count":1,"subscribers_count":2,"default_branch":"master","last_synced_at":"2025-03-11T14:56:59.402Z","etag":null,"topics":["gan"],"latest_commit_sha":null,"homepage":null,"language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":null,"status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/innerlee.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":null,"code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null}},"created_at":"2018-02-09T02:47:55.000Z","updated_at":"2020-06-29T03:26:47.000Z","dependencies_parsed_at":"2022-08-27T20:52:22.282Z","dependency_job_id":null,"html_url":"https://github.com/innerlee/ganbase","commit_stats":null,"previous_names":[],"tags_count":1,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/innerlee%2Fganbase","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/innerlee%2Fganbase/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/innerlee%2Fganbase/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/innerlee%2Fganbase/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/innerlee","download_url":"https://codeload.github.com/innerlee/ganbase/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":247378087,"owners_count":20929291,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["gan"],"created_at":"2024-10-31T09:17:38.312Z","updated_at":"2025-04-05T17:41:43.741Z","avatar_url":"https://github.com/innerlee.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# GAN \n## Image Generation\n### DCGAN \n[Arxiv](https://arxiv.org/pdf/1511.06434.pdf)\n### WGAN\n[Arxiv](https://arxiv.org/pdf/1701.07875.pdf)\n### WGAN-GP\n[Arxiv](https://arxiv.org/pdf/1704.00028.pdf)\n### BEGAN\n[Arxiv](https://arxiv.org/pdf/1703.10717.pdf)\n### SAGAN\n[Arxiv](https://arxiv.org/pdf/1805.08318.pdf)\n\n### Experiments on Cifar-10\n![Image](https://github.com/innerlee/ganbase/raw/lxh2/res/cifar10.jpg)\n### Experiments on ImageNet\n![Image](https://github.com/innerlee/ganbase/raw/lxh2/res/imagenet.jpg)\n### Experiments on CelebA\n![Image](https://github.com/innerlee/ganbase/raw/lxh2/res/celeba.jpg)\n\n## Style Transfer\n### pix2pix\n[Arxiv](https://arxiv.org/pdf/1611.07004.pdf)\n#### pix2pix on Cityscapes\n![Image](https://github.com/innerlee/ganbase/raw/lxh2/res/pix2pix.jpg)\n### CycleGAN\n[Arxiv](https://arxiv.org/pdf/1703.10593.pdf)\n### Attention-CycleGAN\n[Arxiv](https://arxiv.org/pdf/1806.02311.pdf)\n#### CycleGAN vs Attention-CycleGAN on horse2zebra\n![Image](https://github.com/innerlee/ganbase/raw/lxh2/res/horse2zebra.jpg)\n#### CycleGAN vs Attention-CycleGAN on orange2apple\n![Image](https://github.com/innerlee/ganbase/raw/lxh2/res/orange2apple.jpg)\n### StarGAN\n[Arxiv](https://arxiv.org/pdf/1711.09020.pdf)\n#### StarGAN on CelebA\n![Image](https://github.com/innerlee/ganbase/raw/lxh2/res/stargan.jpg)\n### FaderNet\n[Arxiv](https://arxiv.org/pdf/1706.00409.pdf)\n#### FaderNet on CelebA\n![Image](https://github.com/innerlee/ganbase/raw/lxh2/res/fadernet.jpg)\n#### StarGAN vs FaderNet on CelebA\n![Image](https://github.com/innerlee/ganbase/raw/lxh2/res/fadernet_vs_stargan.jpg)\n## Train\n### DCGAN\n```\npython train/dcgan_model.py --gpu=0 --last_epoch=0 --train_batch=16 --resize_choice=1 --load_size=64 --fine_size=64 --display_fre=10 dim=100 --d_input_dim=1 --save_img_rows=6 --save_img_cols=12  loss_choice=dcgan --gp_choice=none --train_dir=./datasets/ --visdom_env=dcgan --save_dir=./checkpoints/dcgan\n```\n### WGAN\n```\npython train/dcgan_model.py --gpu=0 --last_epoch=0 --train_batch=16 --resize_choice=1 --load_size=64 --fine_size=64 --display_fre=10 dim=100 --d_input_dim=1 --save_img_rows=6 --save_img_cols=12  loss_choice=wgan --gp_choice=none --train_dir=./datasets/ --visdom_env=wgan --save_dir=./checkpoints/wgan\n```\n### WGAN-GP\n```\npython train/dcgan_model.py --gpu=0 --last_epoch=0 --train_batch=16 --resize_choice=1 --load_size=64 --fine_size=64 --display_fre=10 dim=100 --d_input_dim=1 --save_img_rows=6 --save_img_cols=12  loss_choice=wgan --gp_choice=wgan-gp --train_dir=./datasets/ --visdom_env=wgan-gp --save_dir=./checkpoints/wgan-gp\n```\n### BEGAN\n```\npython train/began.py --gpu=0 --last_epoch=0 --train_batch=16 --resize_choice=1 --load_size=64 --fine_size=64 --display_fre=10 --g_input_dim=100 --d_input_dim=1 --save_img_rows=6 --save_img_cols=12 --loss_choice=began --gp_choice=none --train_dir=./datasets/ --visdom_env=began --save_dir=./checkpoints/began\n```\n### SAGAN\n```\npython train/sagan.py --gpu=0 --last_epoch=0 --train_batch=16 --resize_choice=1 --load_size=64 --fine_size=64 --display_fre=10 --g_input_dim=100 --d_input_dim=1 --save_img_rows=6 --save_img_cols=12 --loss_choice=wgan --gp_choice=none --train_dir=./datasets/ --visdom_env=sagan --save_dir=./checkpoints/sagan\n```\n### pix2pix\n```\npython train/pix2pix_model.py --gpu=0 --resize_choice=3 --flip=1 --workers=4 --save_img_rows=2 --save_img_cols=4 --g_norm_type=instance --g_lr=2e-4 --g_weightdecay=5e-5 --d_norm_type=instance --d_lr=2e-4 --d_weightdecay=5e-5 --display_fre=40 --save_fre=20 --lambdaI=0.5 --visdom_port=8891 --loss_choice=lsgan --g_resnet_blocks=9 --train_dir=./datasets/cityscapes/train --save_dir=./checkpoints/pix2pix/cityscapes256_lambdaI100_batch4 --train_batch=4 --visdom_env=pix2pix_cityscapes256_lambdaI100_batch4 --g_input_dim=3 --d_input_dim=6\n```\n### CycleGAN\n```\npython train/cyclegan_model.py --gpu=0 --resize_choice=3 --flip=1 --workers=4 --save_img_rows=2 --save_img_cols=4 --g_norm_type=instance --g_lr=2e-4 --g_weightdecay=5e-5 --d_norm_type=instance --d_lr=2e-4 --d_weightdecay=5e-5 --display_fre=40 --save_fre=20 --lambdaA=10 --lambdaB=10 --lambdaI=0.5 --visdom_port=8891 --loss_choice=lsgan --g_resnet_blocks=9 --train_dir=./datasets/horse2zebra/trainA --train_target_dir=./datasets/horse2zebra/trainB --save_dir=./checkpoints/cyclegan/horse2zebra256_lambdaI0 --train_batch=1 --visdom_env=cyclegan_horse2zebra256_lambdaI0 --g_input_dim=3 --d_input_dim=3\n```\n### Attention-CycleGAN\n```\npython train/att_cyclegan_model.py --gpu=0 --resize_choice=3 --flip=1 --workers=4 --save_img_rows=2 --save_img_cols=4 --g_norm_type=instance --g_lr=2e-4 --g_weightdecay=5e-5 --d_norm_type=instance --d_lr=2e-4 --d_weightdecay=5e-5 --display_fre=20 --save_fre=10 --lambdaA=10 --lambdaB=10 --lambdaI=0 --visdom_port=8891 --loss_choice=lsgan --g_resnet_blocks=9 --train_dir=./datasets/horse2zebra/testA --train_target_dir=./datasets/horse2zebra/testB --save_dir=./checkpoints/att-cyclegan/horse2zebra256_maskT0.1 --train_batch=1 --visdom_env=att-cyclegan_horse2zebra256_maskT0.1 --g_input_dim=3 --d_input_dim=3 --mask_tau=0.1 --train_att_epochs=1 --load_when_switch\n```\n### StarGAN\n```\npython train/stargan_model.py --gpu=0 --load_size=128 --fine_size=128 --resize_choice=2 --flip=1 --workers=4 --save_img_rows=2 --save_img_cols=4 --g_lr=1e-4 --d_lr=1e-4 --visdom_port=8891 --loss_choice=wgan --g_resnet_blocks=6 --d_resnet_blocks=6 --train_dir=./datasets/celeba/Img/img_align_celeba --celeba_attr_path=./datasets/celeba/Anno/list_attr_celeba.txt --selected_attrs Black_Hair Blond_Hair Brown_Hair Male Young --save_dir=./checlpoints/stargan --train_batch=16 --visdom_env=stargan --epochs=150 --save_fre=5 --display_fre=10 --g_lr_decay_gamma=0.999 --d_lr_decay_gamma=0.999 --g_lr_decay_step=1 --d_lr_decay_step=1\n```\n### FaderNet\n```\npython train/fadernet_model.py --gpu=0 --save_dir=./checkpoints/fadernet --train_dir=./datasets/celeba/Img/img_align_celeba --celeba_attr_path=./datasets/celeba/Anno/list_attr_celeba.txt --img_sz=256 --img_fm=3 --instance_norm=False --init_fm=32 --max_fm=512 --n_layers=6 --n_skip=0 --deconv_method=\"convtranspose\" --hid_dim=512 --dec_dropout=0 --lat_dis_dropout=0.3 --n_lat_dis=1 --n_ptc_dis=0 --n_clf_dis=0 --smooth_label=0.2 --lambda_ae=1 --lambda_lat_dis=0.0001 --lambda_ptc_dis=0 --lambda_clf_dis=0 --lambda_schedule=500000 --clip_grad_norm=5 --display_fre=10 --save_fre=1 --train_batch=32 --selected_attrs Young --attr Young --visdom_env=fadernet_young\n```\n## Test\n### DCGAN / WGAN / WGAN-GP / BEGAN / SAGAN\n```\npython test/test_image_generation.py \n```\n### pix2pix / CycleGAN / Attention-CycleGAN\n```\npython test/test_style_transfer.py \n```\n### StarGAN\n```\npython test/test_stargan.py\n```\n### FaderNet\n```\npython test/test_fadernet.py\n```\n\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Finnerlee%2Fganbase","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Finnerlee%2Fganbase","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Finnerlee%2Fganbase/lists"}