在使用自己的图片做数据集时需要做相应的处理(缩/放)
因此在使用CNN等模型之前,可以使用该方法获得你想要的大小的图片数据
import matplotlib.pyplot as plt
import tensorflow as tf
import numpy as np
image_raw_data = tf.gfile.GFile('C:/Users/DELL/Desktop/CWTPic/train1.jpg','r').read() #加载原始图像
with tf.Session() as sess:
img_data = tf.image.decode_jpeg(image_raw_data) #解码
plt.imshow(img_data.eval())
plt.show()
resized = tf.image.resize_images(img_data, [64,64],method=0) #第一个参数为原始图像,第二个参数为图像大小,第三个参数给出了指定的算法
resized = np.asarray(resized.eval(),dtype='uint8') #变为uint8才能显示
plt.imshow(resized)
plt.show()