# -*- coding: UTF-8 -*-import numpy as npimport osfrom scipy.misc import imread, imresizeimport matplotlib.pyplot as pltfrom glob import glob# 读取目录下所有的jpg图片def load_image(image_path, image_size):file_name=glob(image_path+"/*jpg")sample = []for file in file_name:pic = imread(file).astype(np.float32)pic = imresize(pic, (image_size, image_size)).astype(np.float32)sample.append(pic)sample = np.array(sample)return sampleif __name__=='__main__':samples=load_image("./images",150)# 显示第一张图片pic=samples[:1,:,:,:]pic=np.reshape(pic,(150,150,3)).astype(np.uint8)plt.imshow(pic)plt.show()
