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

# -*- coding: UTF-8 -*-
import numpy as np
import os
from scipy.misc import imread, imresize
import matplotlib.pyplot as plt
from 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()