【计算机视觉】批量图片灰度处理方法

摘要

对图片进行批量灰度处理，可以使用 Python 中的 PIL 库来实现。PIL 库是 Python Imaging Library 的缩写，它提供了图像处理的基本功能，包括图像的读取、写入、格式转换、缩放、裁剪和色彩转换等。

实现方法

from PIL import Image
import os

# 定义图片目录
img_dir = 'path/to/image/dir'

# 遍历目录下的所有文件
for filename in os.listdir(img_dir):
    # 判断是否是图片文件
    if filename.endswith('.jpg') or filename.endswith('.png'):
        # 打开图片
        img_path = os.path.join(img_dir, filename)
        with Image.open(img_path) as img:
            # 转换为灰度图像
            # s转换为三通道灰度图 img.convert('RGB').convert('L') 
            gray_img = img.convert('L')
            # 保存灰度图像
            gray_img_path = os.path.join(img_dir, 'gray_' + filename)
            gray_img.save(gray_img_path)

以上代码使用 PIL 库中的 Image.open 方法打开指定目录下所有的图片文件，然后使用 convert 方法将图片转换为灰度图像，最后使用 save 方法将灰度图像保存到和原始图片相同的目录下，文件名加上前缀 “gray_”。需要注意的是，这里将图片转换为灰度图时使用的是 convert('L') 方法，表示将图像转为单通道的灰度图像。如果想将图像转换为三通道的灰度图像，可以使用 convert('RGB').convert('L') 方法

自己训练目标检测模型时需要进行批量图片处理，亲测有效！