import cv2
import numpy as np
import os
from PIL import Image,ImageEnhance
from PIL import Image
import os
folder_path = 'C:/workspace/le-yolo/data/images/train'
for filename in os.listdir(folder_path):
    if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')):
        img_path = os.path.join(folder_path, filename)
        img = Image.open(img_path)
        gray_img = img.convert('L')
        new_filename = os.path.splitext(filename)[0] + '_gray.jpg'  # 修改为新文件名，如 "image_gray.jpg"
        new_img_path = os.path.join(folder_path, new_filename)
        gray_img.save(new_img_path)
        print(f"已将 {filename} 转换为灰度图像并保存")