灰度图转为标签图，0和1像素值转为0和255

from PIL import Image
import os

# 源文件夹路径
folder_path = r'E:\data\1'

# 新文件夹路径
new_folder_path = r'E:\data\2'

# 创建新文件夹（如果不存在）
os.makedirs(new_folder_path, exist_ok=True)

# 遍历源文件夹中的所有内容
for root, dirs, files in os.walk(folder_path):
    # 创建目标文件夹（包括所有中间目录）
    target_folder = root.replace(folder_path, new_folder_path)
    os.makedirs(target_folder, exist_ok=True)

    # 处理每个文件
    for file in files:
        # 读取原始标签图像
        src_file = os.path.join(root, file)
        img = Image.open(src_file)

        # # 将0和1像素值替换为0和255像素值
        img = img.point(lambda p: p * 255)
        # 将0和1像素值替换为255和0像素值
        # img = img.point(lambda p: 255 if p == 0 else 0)

        # 保存到新文件夹
        dst_file = os.path.join(target_folder, file)
        img.save(dst_file)

        print(f'已处理文件: {src_file} -> {dst_file}')

print('所有标签文件已处理完成。')