工业检测中的雨线去除装置、雨线去除方法

摘要

本发明提供一种工业检测中的雨线去除装置、雨线去除方法，涉及工业检测技术领域，装置包括：串联的第一至第三通道；其中，第一至第三通道包括：通道记忆块和特征筛选模块，通道记忆块包括：第一卷积层、通道注意力模块和长短期记忆网络，第一卷积层用于提取特征并转化为特征图，通道注意力模块用于基于上下文语义提取通道上的语义信息，长短期记忆网络用于对语义信息进行筛选。本发明采用渐进式去雨结构，每次都是基于上一次提取结果进一步进行特征提取，解决了单阶段去雨效果较差的问题，利用通道注意力模块进行特征提取，可以对样本数据进行高效的特征提取与传递，直到能够准确提取出需要去除的雨线信息。