python批量缩放文件夹中的图片（图片宽高比不变）

当我们用自己的数据集训练图像分类/目标检测模型时，常常需要先将数据集中的图片缩小到适当的尺寸。尤其是用自己拍摄的图片做数据集时，图片尺寸往往过大，这时需要按原图宽高比批量缩放图片。

话不多说，直接上代码：

import os
from PIL import Image
 
filename = os.listdir("./aaaaa")  #原图片集所在目录
base_dir = "./aaaaa/"
new_dir  = "./bbbbb/"    #缩放后的图片存放的路径
new_longest = 500         #设置新图像最长边像素数
 
for img in filename:      #遍历文件夹中的图片
    image = Image.open(base_dir + img)
    width =image.size[0]   #原图宽
    height =image.size[1]  #原图高
    #以下代码实现按设置好的最长边像素数缩放图片，并保持宽高比不变
    if(width>=height):
        longest=width
        new_height = int(height * new_longest / width)
        out=image.resize((new_longest,new_height),Image.ANTIALIAS)
    else:
        longest=height
        new_width = int(width * new_longest / height)
        out=image.resize((new_width,new_longest),Image.ANTIALIAS)
    out.save(new_dir+ img)   #保存新图片