彩色转灰度图

from PIL import Image
import os
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True


def convert_images_to_grayscale(folder_path, output_folder_path):
    # 确保输出文件夹存在
    if not os.path.exists(output_folder_path):
        os.makedirs(output_folder_path)

        # 遍历文件夹中的所有文件
    for filename in os.listdir(folder_path):
        img_path = os.path.join(folder_path, filename)

        # 检查文件是否为图片（这里简单检查文件扩展名，你可以根据需要添加更多检查）
        if img_path.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.tiff')):
            with Image.open(img_path) as img:
                # 转换为灰度图
                gray_img = img.convert('L')

                # 构造输出文件的路径和名称
                output_path = os.path.join(output_folder_path, filename)

                # 保存灰度图
                gray_img.save(output_path)

            # 使用示例


folder_path = r'C:\Users\Administrator\Desktop\yolov5-master\data\1'  # 你的彩色图片文件夹路径
output_folder_path = r'C:\Users\Administrator\Desktop\yolov5-master\data\2'  # 灰度图片保存的文件夹路径
convert_images_to_grayscale(folder_path, output_folder_path)