在做ai项目的时候，比如图像分类和目标检测，数据里面可能有rgb图和灰度图，但是灰度图的通道数只有1，而rgb图的通道数有3。
因此要进行数据预处理，下面代码就可以删除文件夹（包括子文件夹）里面所有非RGB的图像，来使得我们的训练和推理正常进行。

import os
from PIL import Image
import time
def delete_non_rgb_grayscale_images(folder_path):for root, dirs, files in os.walk(folder_path):for filename in files:if filename.endswith('.jpg') or filename.endswith('.jpeg'):file_path = os.path.join(root, filename)try:img = Image.open(file_path)if img.mode != 'RGB':print(f"Deleting {file_path} as it is not in RGB mode.")img.close()  # 关闭图像对象，这个非常重要，要不然会无法进行删除文件操作os.unlink(file_path)  # 尝试强制删除文件except Exception as e:print(f"Error processing {file_path}: {str(e)}")# 指定包含图片的根文件夹路径
folder_path = r"path\AllClasses"
delete_non_rgb_grayscale_images(folder_path)