【计算机视觉】批量图片灰度处理方法
自己训练目标检测模型时需要进行批量图片处理,亲测有效!
·
摘要
对图片进行批量灰度处理,可以使用 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') 方法
自己训练目标检测模型时需要进行批量图片处理,亲测有效!
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐

所有评论(0)