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

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

from PIL import Image
import osdef 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)