在 /home/your_name/TensorFlow/DCGAN/ 下新建文件 train.py，同時新建文件夾 logs 和文件夾 samples，前者用來保存訓練過程中的日志和模型，后者用來保存訓練過程中采樣器的采樣圖片，在 train.py 中輸入如下代碼：
# -*- coding: utf-8 -*-import tensorflow as tfimport osfrom read_data import *from utils import *from ops import *from model import *from model import BATCH_SIZEdef train(): # 設(shè)置 global_step ，用來記錄訓練過程中的 step
global_step = tf.Variable(0, name = 'global_step', trainable = False) # 訓練過程中的日志保存文件
train_dir = '/home/your_name/TensorFlow/DCGAN/logs'
# 放置三個 placeholder，y 表示約束條件，images 表示送入判別器的圖片，
# z 表示隨機噪聲
y= tf.placeholder(tf.float32, [BATCH_SIZE, 10], name='y')
images = tf.placeholder(tf.float32, [64, 28, 28, 1], name='real_images')
z = tf.placeholder(tf.float32, [None, 100], name='z') # 由生成器生成圖像 G
G = generator(z, y) # 真實圖像送入判別器
D, D_logits = discriminator(images, y) # 采樣器采樣圖像
samples = sampler(z, y) # 生成圖像送入判別器
D_, D_logits_ = discriminator(G, y, reuse = True)
延伸閱讀
- ssh框架
2016-09-30
- 阿里移動安全 [無線安全]玩轉(zhuǎn)無線電——不安全的藍牙鎖
2017-07-26
- 消息隊列NetMQ 原理分析4-Socket、Session、Option和Pipe
2024-03-26
- Selective Search for Object Recognition 論文筆記【圖片目標分割】
2017-07-26
- 詞向量-LRWE模型-更好地識別反義詞同義詞
2017-07-26
- 從棧不平衡問題 理解 calling convention
2017-07-26
- php imagemagick 處理 圖片剪切、壓縮、合并、插入文本、背景色透明
2017-07-26
- Swift實現(xiàn)JSON轉(zhuǎn)Model - HandyJSON使用講解
2017-07-26
- 阿里移動安全 Android端惡意鎖屏勒索應用分析
2017-07-26
- 集合結(jié)合數(shù)據(jù)結(jié)構(gòu)來看看(二)
2017-07-26
學習是年輕人改變自己的最好方式