import tensorflow as tf

## 读入
image_raw_data = tf.gfile.FastGFile("F:/datasets/cup.jpg", 'rb').read()
img_data = tf.image.decode_jpeg(image_raw_data)  # unit8格式

resized = tf.image.resize_images(img_data, [300, 300], method=0)  # float32 格式

sess = tf.Session()
import matplotlib.pyplot as plt
plt.imshow(sess.run(resized/255)) # 因为是float格式，所以显示时应当在0~1之间
plt.show()

# 写出
retype = tf.cast(resized, tf.uint8)
encoder_image = tf.image.encode_png(retype)
with tf.gfile.GFile("F:/output.png", 'wb') as f:
    f.write(sess.run(encoder_image))
sess.close()