-
大小: 37.97MB文件類型: .zip金幣: 1下載: 0 次發布日期: 2023-07-24
- 語言: Python
- 標簽:
資源簡介
利用TensorFlow中的深度學習進行圖像修復
代碼片段和文件信息
#!/usr/bin/env?python3
#
#?Brandon?Amos?(http://bamos.github.io)
#?License:?MIT
#?2016-08-05
import?argparse
import?os
import?tensorflow?as?tf
from?model?import?DCGAN
parser?=?argparse.ArgumentParser()
parser.add_argument(‘--approach‘?type=str
????????????????????choices=[‘adam‘?‘hmc‘]
????????????????????default=‘adam‘)
parser.add_argument(‘--lr‘?type=float?default=0.01)
parser.add_argument(‘--beta1‘?type=float?default=0.9)
parser.add_argument(‘--beta2‘?type=float?default=0.999)
parser.add_argument(‘--eps‘?type=float?default=1e-8)
parser.add_argument(‘--hmcBeta‘?type=float?default=0.2)
parser.add_argument(‘--hmcEps‘?type=float?default=0.001)
parser.add_argument(‘--hmcL‘?type=int?default=100)
parser.add_argument(‘--hmcAnneal‘?type=float?default=1)
parser.add_argument(‘--nIter‘?type=int?default=1000)
parser.add_argument(‘--imgSize‘?type=int?default=64)
parser.add_argument(‘--lam‘?type=float?default=0.1)
parser.add_argument(‘--checkpointDir‘?type=str?default=‘checkpoint‘)
parser.add_argument(‘--outDir‘?type=str?default=‘completions‘)
parser.add_argument(‘--outInterval‘?type=int?default=50)
parser.add_argument(‘--maskType‘?type=str
????????????????????choices=[‘random‘?‘center‘?‘left‘?‘full‘?‘grid‘?‘lowres‘]
????????????????????default=‘center‘)
parser.add_argument(‘--centerScale‘?type=float?default=0.25)
parser.add_argument(‘imgs‘?type=str?nargs=‘+‘)
args?=?parser.parse_args()
assert(os.path.exists(args.checkpointDir))
config?=?tf.ConfigProto()
config.gpu_options.allow_growth?=?True
with?tf.Session(config=config)?as?sess:
????dcgan?=?DCGAN(sess?image_size=args.imgSize
??????????????????batch_size=min(64?len(args.imgs))
??????????????????checkpoint_dir=args.checkpointDir?lam=args.lam)
????dcgan.complete(args)
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2017-07-09?21:32??dcgan-completion.tensorflow-master\
?????文件??????????44??2017-07-09?21:32??dcgan-completion.tensorflow-master\.gitignore
?????文件????????1456??2017-07-09?21:32??dcgan-completion.tensorflow-master\LICENSE
?????文件????????1429??2017-07-09?21:32??dcgan-completion.tensorflow-master\README.md
?????目錄???????????0??2017-07-09?21:32??dcgan-completion.tensorflow-master\checkpoint\
?????文件????37822480??2017-07-09?21:32??dcgan-completion.tensorflow-master\checkpoint\DCGAN.model-60502.data-00000-of-00001
?????文件????????1813??2017-07-09?21:32??dcgan-completion.tensorflow-master\checkpoint\DCGAN.model-60502.index
?????文件?????1599466??2017-07-09?21:32??dcgan-completion.tensorflow-master\checkpoint\DCGAN.model-60502.me
?????文件??????????91??2017-07-09?21:32??dcgan-completion.tensorflow-master\checkpoint\checkpoint
?????文件????????1791??2017-07-09?21:32??dcgan-completion.tensorflow-master\complete.py
?????文件?????4627254??2017-07-09?21:32??dcgan-completion.tensorflow-master\completion.compressed.gif
?????文件???????20515??2017-07-09?21:32??dcgan-completion.tensorflow-master\model.py
?????文件????????4416??2017-07-09?21:32??dcgan-completion.tensorflow-master\ops.py
?????文件????????1002??2017-07-09?21:32??dcgan-completion.tensorflow-master\simple-distributions.py
?????文件????????1641??2017-07-09?21:32??dcgan-completion.tensorflow-master\train-dcgan.py
?????文件????????7217??2017-07-09?21:32??dcgan-completion.tensorflow-master\utils.py
評論
共有 條評論