去雾去雨算法

简单版

import cv2
import numpy as np
 
def dehaze(image):
    """简单去雾算法，使用直方图均衡化来增强图像"""
    # 将图像转换为YUV颜色空间
    yuv_image = cv2.cvtColor(image, cv2.COLOR_BGR2YUV)
 
    # 对Y通道（亮度）进行直方图均衡化
    yuv_image[:, :, 0] = cv2.equalizeHist(yuv_image[:, :, 0])
 
    # 将YUV图像转换回BGR
    dehazed_image = cv2.cvtColor(yuv_image, cv2.COLOR_YUV2BGR)
 
    return dehazed_image
 
def derain(image):
    """简单去雨算法，使用高通滤波器去除细小的雨滴噪声"""
    # 转换为灰度图像
    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
 
    # 使用高通滤波器去除雨滴
    high_pass_filter = cv2.GaussianBlur(gray_image, (21, 21), 0)
    derained_image = cv2.subtract(gray_image, high_pass_filter)
 
    # 增强对比度
    derained_image = cv2.equalizeHist(derained_image)
 
    # 将灰度图转换回BGR格式
    derained_image = cv2.cvtColor(derained_image, cv2.COLOR_GRAY2BGR)
 
    return derained_image
 
# 加载图像
image = cv2.imread('