图像预处理（单通道化图像）

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位，则图片是单通道图片