数据增强：图片加雾效果实现Python

图像处理中多数算法为去雾算法，所以写了个加雾的算法，加雾效果如下：

直接请上代码：

import cv2, math
import numpy as np

def demo():
    img_path = 'test.png'

    img = cv2.imread(img_path)
    img_f = img / 255.0
    (row, col, chs) = img.shape

    A = 0.5                               # 亮度
    beta = 0.08                           # 雾的浓度
    size = math.sqrt(max(row, col))      # 雾化尺寸
    center = (row // 2, col // 2)        # 雾化中心
    for j in range(row):
        for l in range(col):
            d = -0.04 * math.sqrt((j-center[0])**2 + (l-center[1])**2) + size
            td = math.exp(-beta * d)
            img_f[j][l][:] = img_f[j][l][:] * td + A * (1 - td)

    cv2.imshow("src", img)
    cv2.imshow("dst", img_f)
    cv2.waitKey()


if __name__ == '__main__':
    demo()