一种基于高效卷积与注意力机制的轻量化钢铁表面缺陷分割模型

本发明公开了一种基于高效卷积与注意力机制的轻量化钢铁表面缺陷分割模型，属于机器视觉领域。本发明主要包括以下模块：轻量级特征提取网络，轻量化特征增强模块，层级自适应上采样融合模块，轻量化分割头模块。本发明模型在特征提取阶段，采用轻量级特征提取模块用来高效提取空间细节信息以及语义信息，然后在解码器阶段通过轻量化局部特征增强模块对特定尺度的特征进一步增强，并选取1/4原图大小的特征图做空间细节方面的特征增强，之后将增强后的特征通过层级自适应上采样融合模块进行融合，以恢复原始的特征。该方法可以有效分割缺陷图像，在实时性和准确性之前达到很好的平衡。为实际场景下的钢铁表面缺陷分割提供新的解决方案。