一种基于离散注意力机制的图像去雨方法及系统

    公开(公告)号:CN116994191A

    公开(公告)日:2023-11-03

    申请号:CN202310251085.X

    申请日:2023-03-16

    Inventor: 罗岩 周杰

    Abstract: 本发明公开了图像去雨领域的一种基于离散注意力机制的图像去雨方法及系统,包括:采集监测区域的实时监测图像,将实时监测图像输入至训练好的信息蒸馏去雨网络模型,对实时监测图像去除雨纹特征获得输出图像;所述信息蒸馏去雨网络模型的训练过程包括:将所述多空间卷积池化网络模块、长短期记忆网络模块和离散注意力信息蒸馏网络模块串联,构建信息蒸馏去雨网络模型;利用训练样本集S对信息蒸馏去雨网络模型进行训练并输出训练结果;利用损失函数对信息蒸馏去雨网络模型的参数进行优化迭代,重复迭代直至达到设定的最大迭代次数输出训练好的信息蒸馏去雨网络模型;在高分辨率下提取更加精细的雨纹特征,从而去除图像中的雨纹特征。

Patent Agency Ranking