雨天交通车辆图像去雨方法、装置、设备及可读存储介质

摘要

本发明公开了雨天交通车辆图像去雨方法、装置、设备及可读存储介质，自然环境光下，雨纹的大小和方向具有多样性，现有的单幅图像去雨算法采用普通卷积提取雨纹特征，特征图的感受野随着卷积层数的增加而呈线性增加，导致提取的细节信息与边缘信息丢失严重，本发明首先采用膨胀卷构造多尺度膨胀卷积融合模块，使不同大小和方向的雨纹特征取得不同的感受野，从而使得提取的特征信息更加丰富且不损失细节信息，其次引入注意力机制构造注意机制残差块，提取的注意特征能较好的实现复原图像的同时，能够将原来偏暗的背景图像中目标边缘特征增强，本发明方法可以有效的去除雨纹，避免丢失细节信息，为后续目标检测等计算机视觉任务奠定良好的基础。