AOD图像去雾python

AOD（Atmospheric Optical Depth）是一种常用的图像去雾方法，可以通过估计图像中的大气光照和透射率来去除雾霾。下面是使用Python实现AOD图像去雾的步骤：

1. 读取原始图像并将其转换为灰度图像。
2. 估计图像中的大气光照。
3. 估计图像中的透射率。
4. 根据估计的大气光照和透射率对原始图像进行去雾处理。

以下是一个简单的Python代码示例，演示如何使用AOD方法对图像进行去雾处理：

import cv2
import numpy as np

def estimate_atmospheric_light(img, p=0.1):
    """
    估计图像中的大气光照
    """
    h, w = img.shape[:2]
    num_pixels = int(w * h * p)
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    dark_channel = np.min(gray, axis=2)
    flat_dark = dark_channel.flatten()
    flat_dark.sort()
    atmospheric_light = np.mean(flat_dark[-num_pixels:])
    return atmospheric_light

def estimate_transmission(img, atmospheric_light, omega=0.95, radius=40):
    """
    估计图像中的透射率
    """
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    transmission = 1 - omega * np.min(gray / atmospheric_light, axis=2)
    kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (radius, radius))
    transmission = cv2.morphologyEx(transmission, cv2.MORPH_CLOSE, kernel)
    return transmission

def dehaze(img, t, atmospheric_light, t0=0.1):
    """
    对原始图像进行去雾处理
    """
    img = img.astype(np.float64)
    atmospheric_light = np.array([atmospheric_light, atmospheric_light, atmospheric_light])
    t = np.maximum(t, t0)
    result = np.empty_like(img)
    for i in range(3):
        result[:, :, i] = (img[:, :, i] - atmospheric_light[i]) / t + atmospheric_light[i]
    result = np.clip(result, 0, 255).astype(np.uint8)
    return result

# 读取原始图像
img = cv2.imread('input.jpg')

# 估计大气光照
atmospheric_light = estimate_atmospheric_light(img)

# 估计透射率
transmission = estimate_transmission(img, atmospheric_light)

# 对原始图像进行去雾处理
result = dehaze(img, transmission, atmospheric_light)

# 显示结果
cv2.imshow('Input', img)
cv2.imshow('Output', result)
cv2.waitKey(0)
cv2.destroyAllWindows()