将CLAHE添加到数据增强中最简单的方法是什么？

import cv2
import numpy as np

def apply_clahe(image):
    # 将图像转换为灰度图像
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    
    # 创建CLAHE对象
    clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8, 8))
    
    # 应用CLAHE到灰度图像
    enhanced = clahe.apply(gray)
    
    # 将增强后的图像转换回彩色图像
    enhanced = cv2.cvtColor(enhanced, cv2.COLOR_GRAY2BGR)
    
    return enhanced

# 假设有一个图像数据集，存储在images列表中
augmented_images = []

for image in images:
    # 应用CLAHE到图像
    enhanced_image = apply_clahe(image)
    
    # 将增强后的图像添加到增强图像列表中
    augmented_images.append(enhanced_image)