python批量将文件夹中的灰度图转为RGB彩色图

注意修改这两个路径，指定包含灰度图的文件夹和输出文件夹

input_folder = r"C:\dyh\ultralytics\yolov5-master\yolov5-master\data_detect_lunxing\val\images"
output_folder = r"C:\dyh\ultralytics\yolov5-master\yolov5-master\origin_bmp\RGB"

from PIL import Image
import os

# 指定包含灰度图的文件夹和输出文件夹
input_folder = r"C:\dyh\ultralytics\yolov5-master\yolov5-master\data_detect_lunxing\val\images"
output_folder = r"C:\dyh\ultralytics\yolov5-master\yolov5-master\origin_bmp\RGB"

# 确保输出文件夹存在
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 遍历输入文件夹中的所有文件
for file_name in os.listdir(input_folder):
    if file_name.endswith(".bmp") or file_name.endswith(".jpg"):
        # 读取灰度图
        gray_image = Image.open(os.path.join(input_folder, file_name)).convert("L")

        # 转换为 RGB 图
        rgb_image = gray_image.convert("RGB")

        # 生成输出文件路径
        output_file_path = os.path.join(output_folder, file_name)

        # 保存为 RGB 图
        rgb_image.save(output_file_path)

print("转换完成！")