Python-利用TensorFlow中的深度学习进行图像修复

利用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.meta
     文件          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