Feb, 2024

联合注意力引导特征融合网络用于表面缺陷的显著性检测

TL;DR面部缺陷检查在工业生产过程中起着重要作用。尽管基于卷积神经网络(CNN)的缺陷检查方法取得了巨大进展,但仍然面临许多挑战,如缺陷尺度变化,复杂背景,低对比度等。为了解决这些问题,我们提出了一个基于编码器 - 解码器网络的联合注意力引导特征融合网络(JAFFNet)来进行表面缺陷的显著性检测。JAFFNet 主要将联合注意力引导特征融合(JAFF)模块并入解码阶段,以自适应地融合低级和高级特征。JAFF 模块学习强调缺陷特征,并在特征融合过程中抑制背景噪声,有益于检测低对比度的缺陷。此外,JAFFNet 在编码器之后引入了一个密集感受野(DRF)模块,用于捕捉具有丰富上下文信息的特征,有助于检测不同尺度的缺陷。JAFF 模块主要利用高级语义特征提供的学习联合通道 - 空间注意力图来引导特征融合。DRF 模块利用一系列多感受野(MRF)单元,每个单元将前面的所有 MRF 特征图和原始输入作为输入。得到的 DRF 特征捕捉了具有大范围感受野的丰富上下文信息。在 SD-saliency-900、磁砖和 DAGM 2007 上进行的大量实验表明,我们的方法在与其他最先进的方法进行比较时取得了有希望的性能。与此同时,我们的方法实现了每秒 66 帧的实时缺陷检测速度。