TensorFlow读取图片tf.gfile.FastGFile

tf.gfile.FastGFile（）

功能：读取图像

参数：1、图像路径；2、读取方式（‘r’为utf-8，‘rb’为非utf-8）

import tensorflow as tf 
import matplotlib.pyplot as plt

#读取图像
src_img = tf.gfile.FastGFile('lena.jpg', 'rb').read()

with tf.Session() as sess:
    #图像解码
    img_after_decoder = tf.image.decode_jpeg(src_img)

    #显示图像
    print(img_after_decoder.eval())
    plt.imshow(img_after_decoder.eval())
    plt.show()

    #图像编码
    encode_image = tf.image.encode_jpeg(img_after_decoder)
    #保存图像至本地文件
    with tf.gfile.GFile('encode_image.jpg', 'wb') as f:
        f.write(encode_image.eval())