数据集文件提取

  • Post author:
  • Post category:其他



tf.gfile.FastGFile(‘path’, ‘rb’).read()


读取结果是最原始的图像,没有经过解码。如果要显示读入的图像,则需要解码,tf.image.decode_jepg和tf.image.decode_png分别用于解码jpg格式和png格式的图像,得到图像的像素值,这个像素值可以用于显示图像。

image_value = tf.read_file('data/train/cat.11.jpg')
img = tf.image.decode_jpeg(image_value, channels=3)
with tf.Session() as sess:
  print(type(image_value)) # bytes
  print(type(img)) # Tensor
  print(type(img.eval())) # numpy.ndarray !!!
  print(img.eval().shape)#(410,431,3)
  print(img.eval().dtype)#unit8
  plt.figure(1)
  plt.imshow(img.eval())
  plt.show()

decode输出是Tensor,eval后是ndarray

读取文件名字然后获取图像信息

将图像文件名及标签放入队列中


input_queue=tf.train.slice_input_producer([image,label]) label = input_queue[1]



读取图片的全部信息



image_contents = tf.read_file(input_queue[0])



把图片解码,



版权声明:本文为weixin_41855385原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。