使用神经网络对图片文件进行训练时,需要将图片信息转换为张量,这里介绍如何将图片信息转化为Tensorflow的张量信息。
本文完整代码:https://github.com/iapcoder/TensorflowReadIMG
一 步骤
1、构造文件队列
file_queue = tf.train.string_input_producer(file_list, shuffle=True) #file_list:图片路径列表,shuffle: True表示随机打乱顺序,返回一个队列
2、构造图像阅读器读取图片内容(默认只读取一张图片)
reader = tf.WholeFileReader()
key, value = reader.read(file_queue) # key:文件名 value:文件内容
3、对读取的数据进行解码(不同的图片格式有不同的解码方式),以jpeg格式为例
image = tf.image.decode_jpeg(value)
4、处理图片的大小、使得每个样本的height、weight一样,方便后续处理
image_resize = tf.image.resize_image(image, [100,100]) # 将图片处理成100*100像素