Defect detection plays an important role in the industrial production and manufacturing sectors, directly impacting product quality and production costs. Most defect detection methods relying on RGB images often encounter challenges such as similarity between foreground and background. RGB-D image that introduces the depth channel can combine RGB and depth information to improve the defect detection performance. However, most RGB-D defect detection methods usually suffer from lower-quality depth images and can not fully take advantage of the two modalities. To tackle this problem, we propose an Adaptive Dual Attention Fusion Network (ADAFNet). ADAFNet is a three-branch decoder-encoder structure that extracts RGB and depth features and synchronously fuses them via a dual-attention dual-modality fusion (DADMF) module. DADMF first carries out dual-modality fusion on the channel and spatial attention features, respectively. Then it makes dual-attention fusion on the RGB and depth features, respectively. Finally, DADMF combines the resulting RGB and depth features using an adaptive weighting factor. Experiments conducted on the NEU RSDDS-AUG, NJU2K, and NLPR datasets demonstrate that our method outperforms other state-of-the-art approaches.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Adaptive Dual Attention Fusion Network for RGB-D Surface Defect Detection

  • Xiaoheng Jiang,
  • Jingqi Liu,
  • Feng Yan,
  • Yang Lu,
  • Shaohui Jin,
  • Hao Liu,
  • Mingliang Xu

摘要

Defect detection plays an important role in the industrial production and manufacturing sectors, directly impacting product quality and production costs. Most defect detection methods relying on RGB images often encounter challenges such as similarity between foreground and background. RGB-D image that introduces the depth channel can combine RGB and depth information to improve the defect detection performance. However, most RGB-D defect detection methods usually suffer from lower-quality depth images and can not fully take advantage of the two modalities. To tackle this problem, we propose an Adaptive Dual Attention Fusion Network (ADAFNet). ADAFNet is a three-branch decoder-encoder structure that extracts RGB and depth features and synchronously fuses them via a dual-attention dual-modality fusion (DADMF) module. DADMF first carries out dual-modality fusion on the channel and spatial attention features, respectively. Then it makes dual-attention fusion on the RGB and depth features, respectively. Finally, DADMF combines the resulting RGB and depth features using an adaptive weighting factor. Experiments conducted on the NEU RSDDS-AUG, NJU2K, and NLPR datasets demonstrate that our method outperforms other state-of-the-art approaches.