python read all jpg files in a directory and display the first picture code example
Example: python read all jpg files in a directory and display the first picture
# -*- coding: UTF-8 -*-import numpy as npimport osfrom scipy.misc import imread, imresizeimport matplotlib.pyplot as pltfrom glob import glob # Read all the jpg image directorydef 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) # Display the first picture pic=samples[:1,:,:,:] pic=np.reshape(pic,(150,150,3)).astype(np.uint8) plt.imshow(pic) plt.show()