240627_图像24位深度（RGB图）转为8位深度（单通道图）

240627_图像24位深度（RGB图）转为8位深度（单通道图）

在使用网络上下载下来的一部分图像分割数据集时，有些标签图你看着是一个黑白图，但是他还是有可能是一张RGB三通道图，具体怎么区分呢。右击图片打开属性，打开详细信息，里面可以看到位深度，位深度为24，则为RGB图，位深度为8，则为单通道图。此处就是一个坑，图像分割任务中，标签往往是单通道图，但是有时从网上找到的数据集看起来是黑白的，但是实际训练就会报错，查看了才发现位深度是24，需要用python代码进行修改，以下是具体修改代码：

from PIL import Image
import os


def batch_convert_images(input_folder, output_folder):
    """
    批量将文件夹内的24位深度图片转换为8位深度。

    :param input_folder: 包含图片的输入文件夹路径
    :param output_folder: 转换后图片的输出文件夹路径
    """
    # 确保输出文件夹存在
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # 遍历输入文件夹中的文件
    for filename in os.listdir(input_folder):
        if filename.endswith('.png') or filename.endswith('.png'):  # 检查是否为jpg或jpeg文件
            input_path = os.path.join(input_folder, filename)
            output_path = os.path.join(output_folder, filename)

            # 调用之前的转换函数
            convert_image_depth(input_path, output_path)


def convert_image_depth(input_path, output_path):
    """
    将单张24位深度的图片转换为8位深度。
    """
    with Image.open(input_path) as png:
        img_8bit = png.quantize(colors=256)
        img_8bit.save(output_path)


# 调用批量转换函数
input_folder_path = 'E:\\test\\png24'  # 修改为你的路径
output_folder_path = 'E:\\test\\png8'  # 输出文件夹路径
batch_convert_images(input_folder_path, output_folder_path)