请帮我用python写一个关于水下图像增强的代码 代码要100行以上

好的，这是一段关于水下图像增强的Python代码：

import cv2
import numpy as np

def enhance_underwater_image(img):
    # 将图像转换为YUV颜色空间
    yuv = cv2.cvtColor(img, cv2.COLOR_BGR2YUV)
    # 分离出Y通道
    y_channel = cv2.split(yuv)[0]
    # 计算每个像素的直方图
    hist = cv2.calcHist([y_channel], [0], None, [256], [0, 256])
    # 找到最大峰值
    peak = np.argmax(hist)
    # 对直方图进行归一化
    hist = hist / np.sum(hist)
    # 计算直方图的前缀和
    cdf = np.cumsum(hist)
    # 根据前缀和对图像进行线性映射
    y_channel = np.uint8(255 * cdf[y_channel])
    # 对Y通道进行灰度变换
    y_channel = cv2.equalizeHist(y_channel)
    # 将处理后的Y通道合并回YUV图像
    yuv[..., 0] = y_channel
    # 将YUV图像转换回BGR颜色空间
    result = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR)
    return result

# 读入图像
img = cv2.imread("underwater.jpg")
# 进行水下图像增强
result = enhance_underwater_image(img)
# 保存处理后的图像
cv2.imwrite("enhanced_underwater.jpg", result)

该代码实现了一种基于直方图的水下图像增强算法，具体流程如下：

1. 将图像转换为YUV颜色空间。
2. 分离出Y通道。
3. 计算每个像素的直方图。