一种多尺度高效卷积自注意力单幅图像除雨方法

摘要

本发明公开了一种多尺度高效卷积自注意力单幅图像除雨方法，首先通过图像数据预处理，获得对应的有雨图像和无雨图像，再将所述有雨图像传入融合了改进型Transformer自注意力模块和多尺度空间特征融合模块的网络模型进行迭代训练，经过混合损失函数优化输出与所述无雨图像接近的处理图像，保存训练好的网络模型，再使用训练好的网络模型对需要测试的图像数据完成预测输出除雨后的图像，本方法避免了基于模型驱动的除雨方法存在的泛化能力不足和基于数据驱动的除雨方法产生较大的计算开销等问题，解决了现有技术中的单幅图像除雨方法计算量大且效率偏低的技术问题。