img = Image.open("D:/wjd/5/1.png") # 将图片resize到224x224大小 img = img.resize((im_width, im_height)) #将灰度图转化为RGB模式 img = img.convert("RGB") # 归一化 img1 = np.array(img) / 255. # 将图片增加一个维度,目的是匹配网络模型 img1 = (np.expand_dims(img1, 0)),如何加载一个文件夹啊
时间: 2024-03-27 14:36:14 浏览: 57
基于YOLOV5头部检测+DeepSort的行人跟踪+基于DFLD(MobileNet/ResNet50)性别和年龄检测统计分析
要加载一个文件夹中的所有图片,可以使用Python中的os和glob模块。可以使用以下代码:
```
import os
import glob
from PIL import Image
import numpy as np
folder_path = "D:/wjd/5/"
im_width = 224
im_height = 224
image_list = []
for filename in glob.glob(os.path.join(folder_path, '*.png')):
img = Image.open(filename)
img = img.resize((im_width, im_height))
img = img.convert("RGB")
img = np.array(img) / 255.
img = np.expand_dims(img, 0)
image_list.append(img)
images = np.concatenate(image_list, axis=0)
```
这个代码将会读取`D:/wjd/5/`文件夹下所有的`.png`图片,将它们resize到`224x224`大小,转化为RGB模式,归一化,并将它们存储在一个numpy数组中。最终的数组大小为`(图片数量, 224, 224, 3)`。
阅读全文