工具:pycharm + anaconda
今天在学习《Tensorflow 实战google 深度学习框架》第7章的过程中,在复现 tf.image.resize_images()时发现,当环境配置为tensorflow-gpu时,该函数得到的图像是乱码,与期望的完全不一样,而当将tensorflow换成anaconda中的tensorflow时,一切显示正常。百思不得其解。
具体情况如下:
code:
#-*-coding:utf-8-*-
import matplotlib.pyplot as plt
import tensorflow as tf
filepath = '/media/wzg16/DATA2/Datasets/data/test_data/cat.jpg'
image_raw_data_jpg = tf.gfile.FastGFile(filepath, 'rb').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.float32)
resize_0 = tf.image.resize_images(img_data_jpg, (500, 500), method=0)
resize_1 = tf.image.resize_images(img_data_jpg, (500, 500), method=1)
resize_2 = tf.image.resize_images(img_data_jpg, (500, 500), method=2)
resize_3 = tf.image.resize_images(img_data_jpg, (500, 500), method=3)
print(resize_0.get_shape)
plt.subplot(2,2,1)
plt.imshow(resize_0.eval())
plt.subplot(2, 2, 2)
plt.imshow(resize_1.eval())
plt.subplot(2, 2, 3)
plt.imshow(resize_2.eval())
plt.subplot(2, 2, 4)
plt.imshow(resize_3.eval())
plt.show()
结果:
anaconda + tensorflow 环境下:
可以正确得到结果图:
anaconda + tensorflow-gpu 环境下:
出现以下错误:
1)resize_1 = tf.image.resize_images(img_data_jpg, (500, 500), method=1) 不能正常执行,报错如下:
2)当注释掉1)的相关操作以后,得到以下结果图像:
究竟是什么原因导致的未知,有一篇文章说尽量不要用 tf.image.resize_images() 函数,如下:Tensorflow使用教训:被tf.image.resize浪费的60天