tensorflow读取图片的方法

来源:互联网 发布:长沙软件培训 编辑:程序博客网 时间:2024/05/21 09:13

下面讲解tensorflow如何读取jpg格式的图片,png格式的图片是一样的。有两种情况:

第一种就是把图片看做是一个图片直接读进来,获取图片的原始数据,再进行解码,主要用到的函数就是tf.gfile.FastGFile,tf.image.decode_jpeg

例如:

import tensorflow as tf;  image_raw_data = tf.gfile.FastGFile('/home/penglu/Desktop/11.jpg').read()image = tf.image.decode_jpeg(image_raw_data) #图片解码print image.eval(session=tf.Session())
输出:

[[[ 11  63 110]
  [ 14  66 113]
  [ 17  69 116]
  ..., 

第二种方式就是把图片看看成一个文件,用队列的方式读取

例如:

import tensorflow as tf;  path = '/home/penglu/Desktop/11.jpg'file_queue = tf.train.string_input_producer([path]) #创建输入队列image_reader = tf.WholeFileReader()_, image = image_reader.read(file_queue)image = tf.image.decode_jpeg(image)with tf.Session() as sess:coord = tf.train.Coordinator() #协同启动的线程threads = tf.train.start_queue_runners(sess=sess, coord=coord) #启动线程运行队列print sess.run(image)coord.request_stop() #停止所有的线程coord.join(threads)
输出:

[[[ 11  63 110]
  [ 14  66 113]
  [ 17  69 116]
  ..., 


原创粉丝点击