批量处理图像尺寸

VisionX Lab

于 2021-12-11 16:12:51 发布

阅读量606

点赞数

CC 4.0 BY-SA版权

文章标签：计算机视觉 opencv 人工智能

本文链接：https://blog.youkuaiyun.com/hasque2019/article/details/121875138

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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))