TensorFlow加载图片的方法

Posted 2020-09-29 小淼博客

方法一：直接使用tensorflow提供的函数image = tf.gfile.FastGFile(\'PATH\')来读取一副图片:

import matplotlib.pyplot as plt;  
import tensorflow as tf;  
  
% matplotlib inline #将matplotlib绘制的图像直接输出到当前交互式的框架下

image_raw_data_jpg = tf.gfile.FastGFile(\'home/ubuntu-mm/TensorFlow/Learning/D-TestJupyter/image/Train/Pic.jpg\', \'r\').read()  
  
with tf.Session() as sess:  
    img_data_jpg = tf.image.decode_jpeg(image_raw_data_jpg) #图像解码  
    img_data_jpg = tf.image.convert_image_dtype(img_data_jpg, dtype=tf.uint8) #改变图像数据的类型  
  
    plt.figure(1) #图像显示  
    plt.imshow(img_data_jpg.eval())
    print sess.run(img_data_jpg)

我用的是基于Ubuntu16.04操作系统下的Jupyter来编写的程序，CPU版本（记得加% matplotlib inline就能显示出图像了）

实验的结果如下所示：

方法二：将图像加载到创建好的队列中使用tf.train.string_input_producer（），然后再加载到变量当中：

import tensorflow as tf;    
import matplotlib.pyplot as plt
  
path = \'/home/ubuntu-mm/TesorFlow/Learning/D-TestJupyter/images/Train/Pic.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)  
    image_uint8 = tf.image.convert_image_dtype(image, dtype = tf.uint8)
    plt.imshow(image_uint8.eval())

实验结果如下所示：