批量修改图片大小（适用处理数据集）

       
import os
import cv2
IMG_SIZE = 224
train_dir = os.path.join("./","224_dataset/test")
train_list_dir = os.listdir(train_dir)
train_class_dir = []
for i in range(len(train_list_dir)):
    train_class_dir.append(os.path.join("./","224_dataset/test",train_list_dir[i]))
print(train_class_dir)

num = 0
i = 0
# for i in range(len(train_class_dir)):
while(i<30):
    temp_path = train_class_dir[i]
    img_name_list = os.listdir(temp_path)
    print(img_name_list)
    for single_img_name in img_name_list:
        single_img_path = os.path.join(temp_path,single_img_name)
        img = cv2.imread(single_img_path,cv2.IMREAD_UNCHANGED)
        print("正在处理的图片路径为："+single_img_path)
        img = cv2.resize(img, (IMG_SIZE, IMG_SIZE))
        cv2.imwrite(".//40_224_dataset//test//"+train_list_dir[i]+"//"+single_img_name,img)
        num = num + 1
        print("已处理"+train_list_dir[i]+"类的第"+str(num)+"张") 
    num = 0
    i = i+1

即从“E://224_dataset/test"读取图片

再配合我上一篇博客中批量创建文件夹的方法，创建相同的文件夹。即可完成图片缩放，写入对应的文件中。