加雨(雨天数据集制作）

最新推荐文章于 2025-02-19 10:21:29 发布

geriezmann

最新推荐文章于 2025-02-19 10:21:29 发布

阅读量6.7k

点赞数 11

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/adhkcnbk/article/details/124250435

版权

python 专栏收录该内容

6 篇文章

订阅专栏

这篇博客介绍了如何利用OpenCV库生成随机噪声并应用运动模糊来模拟雨天图像。首先读取图像，然后创建均匀分布的噪声，并通过高斯滤波使其模糊。接着，通过仿射变换和高斯模糊生成雨线，并与原始图像结合，通过透明度赋值或加权合成方法实现雨天效果。最后展示了加权合成的新图像。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

加雨

1、读取图像

```
img = cv2.imread('demo.jpg')
```

2、根据图像生成随机噪声

生成跟图像相同大小的噪声（均匀分布）
```
noise = np.random.uniform(0, 256, img.shape[0:2])
```
控制噪声水平，取浮点数，只保留最大的一部分作为噪声
```
v = value * 0.01
noise[np.where(noise < (256 - v))] = 0
```

噪声做初次模糊（高斯滤波）

k = np.array([[0, 0.1, 0],
              [0.1, 8, 0.1],
              [0, 0.1, 0]])
noise = cv2.filter2D(noise, -1, k)

返回生成的噪声

3、对噪声加运动模糊

根据仿射变换得到模糊核

#获得仿射变化矩阵，由于对角阵自带45度的倾斜，所以减去了45度，三个参数分别表示：中心点的位置，旋转角度，等比例缩放
trans = cv2.getRotationMatrix2D((length/2, length/2), angle-45, 1-length/100.0)
#生成对角矩阵（雨线的长度）
dig = np.diag(np.ones(length))  
#生成模糊核，通过对对角矩阵的仿射变换，得到模糊核，三个参数分别表示为：对角矩阵，仿射变换矩阵，输出的大小
k = cv2.warpAffine(dig, trans, (length, length))

对模糊核进行高斯模糊，使得雨有宽度

k = cv2.GaussianBlur(k, (w, w), 0) #高斯模糊这个旋转后的对角核，使得雨有宽度，高斯矩阵为w*w， 0为方差

将噪声与得到的模糊核进行卷积

blurred = cv2.filter2D(noise, -1, k)
cv2.normalize(blurred, blurred, 0, 255, cv2.NORM_MINMAX)
blurred = np.array(blurred, dtype=np.uint8)

blurred

方法一、透明度赋值

rain_result = img.copy()
rain_result[:, :, 0] = rain_result[:, :, 0] * (255 - rain[:, :, 0])/255.0 + beta*rain[:, :, 0]
rain_result[:, :, 1] = rain_result[:, :, 1] * (255 - rain[:, :, 0])/255.0 + beta*rain[:, :, 0]
rain_result[:, :, 2] = rain_result[:, :, 2] * (255 - rain[:, :, 0])/255.0 + beta*rain[:, :, 0]

加雨的结果图为：

rain_effect

方法二、加权

rain = np.expand_dims(rain, 2)
rain = np.repeat(rain, 3, 2)
#加权合成新图
result = cv2.addWeighted(img, alpha, rain, 1-alpha, 1)

得到的雨天图像

epeat(rain, 3, 2)
#加权合成新图
result = cv2.addWeighted(img, alpha, rain, 1-alpha, 1)


得到的雨天图像

![addweight](https://img-blog.csdnimg.cn/img_convert/411ec1e333f24f045893453a6cb0dd40.png)