当我们用自己的数据集训练图像分类/目标检测模型时,常常需要先将数据集中的图片缩小到适当的尺寸。尤其是用自己拍摄的图片做数据集时,图片尺寸往往过大,这时需要按原图宽高比批量缩放图片。
话不多说,直接上代码:
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) #保存新图片