Sep, 2023

CINFormer: 多阶段 CNN 特征注入的 Transformer 网络用于表面缺陷分割

TL;DR通过使用多阶段 CNN 特征注入的 Transformer 网络,我们提出了一种名为 CINFormer 的 UNet 结构,以实现表面缺陷分割。CINFormer 在编码器中采用了简单而有效的特征整合机制,将输入图像的多层级 CNN 特征注入到 Transformer 网络的不同阶段,以保持 CNN 捕捉详细特征的优点和 Transformer 压制背景噪音的优点,从而实现准确的缺陷检测。此外,CINFormer 通过引入 Top-K 自注意力模块,可以聚焦于与缺陷相关的更重要信息的标记,以进一步减少冗余背景的影响。通过在 DAGM 2007、磁砖和 NEU 等表面缺陷数据集上进行的大量实验证明,CINFormer 在缺陷检测方面达到了最先进的性能。