資源簡介
用pytorch實現的DCGAN,代碼結構清晰,附有說明文件和數據集下載地址。并有結果圖片。下載后請先查看 readme.md文件
代碼片段和文件信息
from?__future__?import?print_function
import?argparse
import?os
import?random
import?time
import?torch
import?torch.backends.cudnn?as?cudnn
import?torch.nn?as?nn
import?torch.nn.parallel
import?torch.optim?as?optim
import?torch.utils.data
import?torchvision.utils?as?vutils
from?torch.autograd?import?Variable
from?data.read_data?import?Get_dataset
from?models.Discriminator?import?_netD
from?models.Generator?import?_netG
parser?=?argparse.ArgumentParser()
parser.add_argument(‘--dataset‘?default=“humanface“help=‘cifar10?|?lsun?|?imagenet?|?folder?|?lfw?|?fake‘)
parser.add_argument(‘--dataroot‘?default=“./face“??help=‘path?to?dataset‘)
parser.add_argument(‘--train_image_list‘default=‘./data/filelist.txt‘help=‘pics?path?lists‘)
parser.add_argument(‘--workers‘?type=int?help=‘number?of?data?loading?workers‘?default=4)
parser.add_argument(‘--batchSize‘?type=int?default=64?help=‘input?batch?size‘)
parser.add_argument(‘--imageSize‘?type=int?default=64?help=‘the?height?/?width?of?the?input?image?to?network‘)
parser.add_argument(‘--nz‘?type=int?default=1000?help=‘size?of?the?latent?z?vector‘)
parser.add_argument(‘--ngf‘?type=int?default=64?help=‘number?of?the?filter?of?generate?network‘)
parser.add_argument(‘--ndf‘?type=int?default=64help=‘number?of?the?filter?of?descriminator?network‘)
parser.add_argument(‘--niter‘?type=int?default=300?help=‘number?of?epochs?to?train?for‘)
parser.add_argument(‘--lr‘?type=float?default=0.0002?help=‘learning?rate?default=0.0002‘)
parser.add_argument(‘--decay_round‘?type=int?default=50?help=‘learning?rate?decay?0.5?each?decay_round‘)
parser.add_argument(‘--beta1‘?type=float?default=0.5?help=‘beta1?for?adam.?default=0.5‘)
parser.add_argument(‘--cuda‘?type=booldefault=True?help=‘enables?cuda‘)
parser.add_argument(‘--ngpu‘?type=int?default=1?help=‘number?of?GPUs?to?use‘)
parser.add_argument(‘--netG‘?default=‘‘?help=“path?to?netG?(to?continue?training)“)
parser.add_argument(‘--netD‘?default=‘‘?help=“path?to?netD?(to?continue?training)“)
parser.add_argument(‘--outpics‘?default=‘./pics‘?help=‘folder?to?output?images‘)
parser.add_argument(‘--outckpts‘?default=‘./checkpoints‘?help=‘folder?to?output?checkpoints‘)
parser.add_argument(‘--outlogs‘?default=‘./logs‘?help=‘folder?to?output?images‘)
parser.add_argument(‘--manualSeed‘?type=int?help=‘manual?seed‘)
def?adjust_learning_rate(optimizers?epoch):
????“““
????Sets?the?learning?rate?to?the?initial?LR?decayed?by?10?every?n?epochs
????“““
????lr?=?opt.lr?*?(0.5?**?(epoch?//?opt.decay_round))
????for?optimizer?in?optimizers:
????????for?param_group?in?optimizer.param_groups:
????????????param_group[‘lr‘]?=?lr
#?custom?weights?initialization?called?on?netG?and?netD
def?weights_init(m):
????classname?=?m.__class__.__name__
????if?classname.find(‘Conv‘)?!=?-1:
????????m.weight.data.normal_(0.0?0.02)
????elif?classname.find(‘BatchNorm‘)?!=?-1:
????????m.weight.data.no
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-01-07?00:05??pytorch?DCGAN\
?????文件????????4998??2018-01-06?23:46??pytorch?DCGAN\README.md
?????目錄???????????0??2018-01-06?23:15??pytorch?DCGAN\data\
?????文件????????1648??2018-01-06?18:56??pytorch?DCGAN\data\HumanFaceDataSet.py
?????文件???????????0??2018-01-06?18:56??pytorch?DCGAN\data\__init__.py
?????目錄???????????0??2018-01-06?23:15??pytorch?DCGAN\data\__pycache__\
?????文件????????1849??2018-01-06?19:16??pytorch?DCGAN\data\__pycache__\HumanFaceDataSet.cpython-35.pyc
?????文件????????1748??2018-01-06?18:56??pytorch?DCGAN\data\__pycache__\HumanFaceDataSet.cpython-36.pyc
?????文件?????????120??2018-01-06?19:16??pytorch?DCGAN\data\__pycache__\__init__.cpython-35.pyc
?????文件?????????125??2018-01-06?18:56??pytorch?DCGAN\data\__pycache__\__init__.cpython-36.pyc
?????文件????????1811??2018-01-06?19:16??pytorch?DCGAN\data\__pycache__\read_data.cpython-35.pyc
?????文件????????1632??2018-01-06?18:56??pytorch?DCGAN\data\__pycache__\read_data.cpython-36.pyc
?????文件???????68362??2018-01-06?18:56??pytorch?DCGAN\data\filelist.txt
?????文件????????2824??2018-01-06?22:09??pytorch?DCGAN\data\read_data.py
?????文件?????2203872??2018-01-06?18:56??pytorch?DCGAN\filelist.txt
?????目錄???????????0??2018-01-06?23:15??pytorch?DCGAN\logs\
?????文件?????4304133??2018-01-06?20:07??pytorch?DCGAN\logs\celebA_64_64_log.txt
?????文件???????10908??2018-01-06?22:08??pytorch?DCGAN\main.py
?????目錄???????????0??2018-01-06?23:15??pytorch?DCGAN\models\
?????文件????????1353??2018-01-06?18:56??pytorch?DCGAN\models\Discriminator.py
?????文件????????1406??2018-01-06?18:56??pytorch?DCGAN\models\Generator.py
?????文件???????????0??2018-01-06?18:56??pytorch?DCGAN\models\__init__.py
?????目錄???????????0??2018-01-06?23:15??pytorch?DCGAN\models\__pycache__\
?????文件????????1365??2018-01-06?19:16??pytorch?DCGAN\models\__pycache__\Discriminator.cpython-35.pyc
?????文件????????1223??2018-01-06?18:56??pytorch?DCGAN\models\__pycache__\Discriminator.cpython-36.pyc
?????文件????????1324??2018-01-06?19:16??pytorch?DCGAN\models\__pycache__\Generator.cpython-35.pyc
?????文件????????1187??2018-01-06?18:56??pytorch?DCGAN\models\__pycache__\Generator.cpython-36.pyc
?????文件?????????122??2018-01-06?19:16??pytorch?DCGAN\models\__pycache__\__init__.cpython-35.pyc
?????文件?????????127??2018-01-06?18:56??pytorch?DCGAN\models\__pycache__\__init__.cpython-36.pyc
?????目錄???????????0??2018-01-06?23:16??pytorch?DCGAN\result\
?????文件??????780608??2018-01-06?23:16??pytorch?DCGAN\result\150.png
............此處省略9個文件信息
評論
共有 條評論