GAN学习 | DCGAN TensorFlow代码解读（1）：main.py

最新推荐文章于 2024-07-26 22:03:17 发布

野狗汪汪汪

最新推荐文章于 2024-07-26 22:03:17 发布

阅读量1k

点赞数 2

本文链接：https://blog.csdn.net/weixin_41274837/article/details/89196355

版权

pprint.PrettyPrinter() 的理解

tf.ConfigProto() 的理解

line 33-105 主函数中定义session

小白总结

代码块学习

Github: DCGAN-Master

本文针对DCGAN的TensorFlow master代码进行解读，并针对MNIST手写数据进行实验。

main.py 代码解读

line1-8 import 库、import自己py文件中的函数和模型

import os
import scipy.misc
import numpy as np
import tensorflow as tf

# 从自定义的文件中import
from model import DCGAN
from utils import pp, visualize, to_json, show_all_variables

line11-30 封装外部传入的参数

flags = tf.app.flags  # 外部参数传递
flags.DEFINE_integer("epoch", 25, "Epoch to train [25]")
flags.DEFINE_float("learning_rate", 0.0002, "Learning rate of for adam [0.0002]")
flags.DEFINE_float("beta1", 0.5, "Momentum term of adam [0.5]")
flags.DEFINE_float("train_size", np.inf, "The size of train images [np.inf]")  # np.inf??
flags.DEFINE_integer("batch_size", 64, "The size of batch images [64]")
flags.DEFINE_integer("input_height", 108, "The size of image to use (will be center cropped). [108]")
flags.DEFINE_integer("input_width", None, "The size of image to use (will be center cropped). If None, same value as input_height [None]")
flags.DEFINE_integer("output_height", 64, "The size of the output images to produce [64]")
flags.DEFINE_integer("output_width", None, "The size of the output images to produce. If None, same value as output_height [None]")
flags.DEFINE_string("dataset", "mnist", "The name of dataset [c