注意:
- 我自己是用来训练医学图像生成新的医学图像的,所以我的label都是1(或者其他数值),因为在GAN训练的时候完全用不到,只有分类的问题才能用到。
- 如果训练分类问题下文中的代码就没用了,不过可以参考这个链接。
- 我的图片是黑白的,所以channel为1,并且图片保存的格式为png的,所以大家理解的时候请注意
image=tf.image.decode_png(image_contents,channels=1)
。
- 代码的输入是png图片,通道数为1,输出为batch_size的
tensor
(shape=[batch_size, image_W, image_H, 1]),但是绝对不能用到feed_dict里面,因为feed_dict只接收numpy格式的数据格式
,所以在接下来的博客中,本博主要在上面下功夫!
tensorflow中 tf.train.slice_input_producer
函数可以参考这个链接,和这个链接讲的非常详细。
- 初始化工作相当重要
输出其中一个样例:
"""
Created on Thu Jul 19 15:40:11 2018
E-mail: Eric2014_Lv@sjtu.edu.cn
@author: DidiLv
"""