当前位置:主页 > 软件编程 > Python代码 >

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

时间:2021-06-16 08:21:27 | 栏目:Python代码 | 点击:

如下所示:

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

您可能感兴趣的文章:

相关文章