使用Python编写水下图像增强的代码

下面是一个简单的Python代码片段，实现了对水下图像的简单增强：

import cv2
import numpy as np

# 读取图像
img = cv2.imread("underwater_image.jpg")

# 通过对色调的修改来增强图像
img_hls = cv2.cvtColor(img, cv2.COLOR_BGR2HLS)
img_hls[:,:,1] = np.where(img_hls[:,:,1] <= 150, img_hls[:,:,1] * 1.5, img_hls[:,:,1])
img = cv2.cvtColor(img_hls, cv2.COLOR_HLS2BGR)

# 保存增强后的图像
cv2.imwrite("enhanced_underwater_image.jpg", img)

代码中使用了OpenCV库来读取图像和进行颜色空间转换，并使用NumPy库来修改图像的色调通道。代码实现了将图像从BGR颜色空间转换为HLS颜色空间，然后对其中的亮度通道进行简单的修改，最后再将图像转换回BGR颜色空间并保存。