python读取目录下所有的jpg文件,并显示第一张图片的示例

# -*- 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 sample if __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()