当我们用自己的数据集训练图像分类/目标检测模型时,常常需要先将数据集中的图片缩小到适当的尺寸。尤其是用自己拍摄的图片做数据集时,图片尺寸往往过大,这时需要按原图宽高比批量缩放图片。
话不多说,直接上代码:
import os
from PIL import Image
filename = os.listdir("./aaaaa") #原图片集所在目录
base_dir = "./aaaaa/"
new_dir = "./bbbbb/" #缩放后的图片存放的路径
new_longest = 500