图像预处理(单通道化图像)

该博客介绍了一个Python函数DataAugment(),用于批量处理图像,将图片库中的图片转换为单通道(灰度)并保存,从而实现数据量的扩大。操作步骤包括读取图片库路径,检查路径是否存在,对每个子目录中的图片进行灰度转换并保存。确保图片位深度为8位,即为单通道图片。此方法适用于数据预处理和图像识别任务的数据增强。
摘要由CSDN通过智能技术生成
Python批量实现图像单通道
函数:DataAugment()
函数功能:扩大数据量
输入参数:dir_path----图片库路径
'''
import cv2
import os
import copy
def DataAugment(dir_path):
    if not os.path.exists(dir_path):
        print('路径不存在')
    else:
        dirs = os.listdir(dir_path)

        for subdir in dirs:
            sub_dir = dir_path + '/' + subdir
            img = cv2.imread(sub_dir, cv2.IMREAD_GRAYSCALE)

            new_name = "%s" % (sub_dir)
            cv2.imwrite(new_name, img)
    print('done')
DataAugment(r"data")
# 地址中不能有中文字符,否则会报错

将批量图片放到./data文件夹下运行程序即可

—————————------------------查看图片是否为单通道图片-------------------————————

右键单击图片,点击属性,点击详细信息,查看图片位深度,如果位深度是8位,则图片是单通道图片

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值