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)