基于选择性注意力机制的图像去雾和去雨算法

摘要

本发明公开了一种基于选择性注意力机制的图像去雾和去雨算法，具体步骤包含如下：扩充训练样本、在训练集上训练网络、对测试图片进行去雾和去雨以及将复原后的图像与清晰图像进行评估三个过程；首先，去雾数据集将所用训练集进行裁剪，截取，旋转等操作进行对原有数据集进行扩充；然后将这些图片输入到基于选择性注意力机制的卷积网络中，使用多损失函数来训练网络；最后同样将测试图片输入到网络中，得到去雾/去雨的结果。本发明实现了一种端到端的去雾/去雨算法，能够高效、准确的对带雨/带雾图像进行去雨/去雾。