Python合成雨图算法

### Python 合成雨图算法实现

在图像处理领域，合成特定效果如雨滴的效果可以通过多种方式来达成。对于Python而言，主要依赖于`OpenCV`和`NumPy`库来进行此类操作。

为了创建下雨效果，在背景图像上随机生成多个透明度渐变的小椭圆模拟雨滴落下过程是一个常见思路[^4]。下面展示一段用于生成带雨水效果的新图像的代码：

import cv2
import numpy as np

def add_rain(image_path, output_path):
    img = cv2.imread(image_path)

    h, w = img.shape[:2]

    # 创建空白画布作为雨层
    rain_layer = np.zeros_like(img)

    num_drops = 1000  # 雨滴数量
    
    for _ in range(num_drops):
        length = np.random.randint(10, 25)
        angle = np.deg2rad(np.random.uniform(-30, 30))
        
        x_start = int(np.random.uniform(w-length))
        y_start = int(np.random.uniform(h))

        dx = int(length * np.cos(angle))
        dy = -int(length * np.sin(angle))

        thickness = np.random.randint(1, 2)
        line_type = 8
        
        color = (255, 255, 255)  # 白色雨线
        
        cv2.line(rain_layer,
                 pt1=(x_start, y_start),
                 pt2=(min(x_start + dx, w), min(y_start + dy, h)),
                 color=color,
                 thickness=thickness,
                 lineType=line_type)

    blended_img = cv2.addWeighted(src1=img, alpha=0.7, src2=rain_layer, beta=0.3, gamma=0)

    cv2.imwrite(output_path, blended_img)


add_rain('input_image.jpg', 'output_with_rain.png')

这段脚本首先读取输入图片并初始化一个同样大小但全黑的画面用来绘制雨滴轨迹；接着通过循环迭代指定次数以增加足够的雨滴数目到场景之中；最后利用加权求和技术将原始照片与人工制造出来的降雨图案融合在一起形成最终效果图。