批量处理图像尺寸

最新推荐文章于 2025-12-10 22:52:50 发布

原创最新推荐文章于 2025-12-10 22:52:50 发布 · 621 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#计算机视觉 #opencv #人工智能

这段代码遍历指定目录中的所有图像，根据文件名前缀创建子目录，并将图像调整为512x512大小。如果图片超过这个尺寸，使用ANTIALIAS选项进行高质量缩放，然后保存到对应标签的目录下。原始图像尺寸小于512则直接复制。该过程有助于组织和优化大量图像数据。

import os
from PIL import Image
import cv2
import shutil

root = '图像路径'
save_path = '处理后的图像保存路径'
for r, d, files in os.walk(root):
    if files != []:
        for i in files:
            fp = os.path.join(r, i)
            label = i.split('_')[0]
            dst = os.path.join(save_path, label)
            if not os.path.exists(dst):
                os.makedirs(dst)

            img = Image.open(fp).convert('RGB')
            w, h = img.size
            if max(w, h) > 512:
                #要处理的图像尺寸，可修改
                img.thumbnail((512, 512), Image.ANTIALIAS)
                img.save(os.path.join(dst, i), quality=95, subsampling=0)
            else:
                shutil.copy(fp, os.path.join(dst, i))