Abstract:No-service rail surface defect (NRSD) segmentation is an essential way for perceiving the quality of no-service rails. However, due to the complex and diverse outlines and low-contrast textures of no-service rails, existing natural image segmentation methods cannot achieve promising performance in NRSD images, especially in some unique and challenging NRSD scenes. To this end, in this paper, we propose a novel segmentation network for NRSDs based on Normalized Attention and Dual-scale Interaction, named NaDiNet. Specifically, NaDiNet follows the enhancement-interaction paradigm. The Normalized Channel-wise Self-Attention Module (NAM) and the Dual-scale Interaction Block (DIB) are two key components of NaDiNet. NAM is a specific extension of the channel-wise self-attention mechanism (CAM) to enhance features extracted from low-contrast NRSD images. The softmax layer in CAM will produce very small correlation coefficients which are not conducive to low-contrast feature enhancement. Instead, in NAM, we directly calculate the normalized correlation coefficient between channels to enlarge the feature differentiation. DIB is specifically designed for the feature interaction of the enhanced features. It has two interaction branches with dual scales, one for fine-grained clues and the other for coarse-grained clues. With both branches working together, DIB can perceive defect regions of different granularities. With these modules working together, our NaDiNet can generate accurate segmentation map. Extensive experiments on the public NRSD-MN dataset with man-made and natural NRSDs demonstrate that our proposed NaDiNet with various backbones (i.e., VGG, ResNet, and DenseNet) consistently outperforms 10 state-of-the-art methods. The code and results of our method are available at <a class="link-external link-https" href="https://github.com/monxxcn/NaDiNet" rel="external noopener nofollow">this https URL</a>.

Mcnet: Multiple Context Information Segmentation Network Of No-Service Rail Surface Defects

No-Service Rail Surface Defect Segmentation via Normalized Attention and Dual-scale Interaction

A Sub-Region Unet for Weak Defects Segmentation with Global Information and Mask-Aware Loss.

A nondestructive automatic defect detection method with pixelwise segmentation

Segmentation of Track Surface Defects Based on Machine Vision and Neural Networks

RSDNet: A New Multiscale Rail Surface Defect Detection Model

Automatic Rail Surface Defects Inspection Based on Mask R-CNN

Hybrid deep learning architecture for rail surface segmentation and surface defect detection

Region and Edge-Aware Network for Rail Surface Defect Segmentation

Two Deep Learning Networks for Rail Surface Defect Inspection of Limited Samples With Line-Level Label

An Adaptive Image Segmentation Network for Surface Defect Detection

Interactive Defect Segmentation in X-Ray Images Based on Deep Learning

A Pixel-Level Segmentation Convolutional Neural Network Based on Global and Local Feature Fusion for Surface Defect Detection

FDSNeT: an Accurate Real-Time Surface Defect Segmentation Network.

YOLOv8n-RSDD: A High-Performance Low-Complexity Rail Surface Defect Detection Network

MRSDI-CNN: Multi-Model Rail Surface Defect Inspection System Based on Convolutional Neural Networks

Normal-Knowledge-Based Pavement Defect Segmentation Using Relevance-Aware and Cross-Reasoning Mechanisms

Multi-modal background-aware for defect semantic segmentation with limited data

AnomalySeg: Deep Learning-Based Fast Anomaly Segmentation Approach for Surface Defect Detection

An Adaptive Feature Refinement Network for Pixel-level Segmentation of Surface Defect

A Rail Surface Defect Detection Method Based on Pyramid Feature and Lightweight Convolutional Neural Network