去雾去雨算法

简单版

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('input_image.jpg')

# 去雾处理
dehazed_image = dehaze(image)

# 去雨处理
derained_image = derain(dehazed_image)

# 保存或展示结果
cv2.imwrite('output_image.jpg', derained_image)
cv2.imshow('Result', derained_image)
cv2.waitKey(0)
cv2.destroyAllWindows()