Surface defect inspection of railways is critical to ensure safe transportation of railway system. However, existing challenges such as irregular defect shapes and similarity between foreground and background hinder surface defect inspection. Previous research methods usually focus on single-level cross-modal fusion, which may not fully utilize multi-modal information. To address this, we introduce a novel framework, termed HM3DNet, which benefits defect detection performance by exploring both hierarchical cross-modal features and depth-enhanced volumetric fusion features. Specifically, we propose a Cross-modal Feature Enhancement (CFE) module to enable the mutual enhancement and complementary fusion of features from the two modalities. To effectively capture rich multi-modal information, we design a Multi-modal Feature Integration (MFI) module to aggregate cross-modal features from CFE and modality-specific features from each decoder. In addition, we design a Depth-Enhanced Volumetric Fusion Network (DeV-FuseNet), which leverages inflated 3D encoder to perform depth-aware fusion of RGB and depth modalities. Moreover, we adopt the dual-stream modality-specific decoder network to capture hierarchical modality-specific information. Experimental results on the NEU RSDDS-AUG dataset revealed that HM3DNet outperformed state-of-the-art (SOTA) methods. We also validate HM3DNet’s generalizability on three other benchmark datasets, where experiment results show that it competes well.

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

Hierarchical Multimodal Feature Learning and 3D Convolution for Rail Defect Detection

  • Shifa Tang,
  • Jinlai Zhang,
  • Shuimiao Yu,
  • Xi Chen,
  • Sheng Wu,
  • Tiefang Zou,
  • Qiqi Li,
  • Lin Hu

摘要

Surface defect inspection of railways is critical to ensure safe transportation of railway system. However, existing challenges such as irregular defect shapes and similarity between foreground and background hinder surface defect inspection. Previous research methods usually focus on single-level cross-modal fusion, which may not fully utilize multi-modal information. To address this, we introduce a novel framework, termed HM3DNet, which benefits defect detection performance by exploring both hierarchical cross-modal features and depth-enhanced volumetric fusion features. Specifically, we propose a Cross-modal Feature Enhancement (CFE) module to enable the mutual enhancement and complementary fusion of features from the two modalities. To effectively capture rich multi-modal information, we design a Multi-modal Feature Integration (MFI) module to aggregate cross-modal features from CFE and modality-specific features from each decoder. In addition, we design a Depth-Enhanced Volumetric Fusion Network (DeV-FuseNet), which leverages inflated 3D encoder to perform depth-aware fusion of RGB and depth modalities. Moreover, we adopt the dual-stream modality-specific decoder network to capture hierarchical modality-specific information. Experimental results on the NEU RSDDS-AUG dataset revealed that HM3DNet outperformed state-of-the-art (SOTA) methods. We also validate HM3DNet’s generalizability on three other benchmark datasets, where experiment results show that it competes well.