• Home
  • Search
  • A Rail Surface Defect Detection Method Based on Pyramid Feature and Lightweight Convolutional Neural Network
  • Cite Icon56
  • https://doi.org/10.1109/tim.2022.3165287Copy DOI Icon

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

  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

The application of computer vision technology in defect detection of industrial products is a popular research direction in recent years. This article presents the pyramid feature convolutional neural network (CNN) for defect detection of rail surfaces. First, multi-scale feature maps are extracted based on the characteristics of defects and backgrounds by the pyramid feature extraction module (PFEM). Then the feature maps are input to a lightweight network consisting of a small number of parameters. The network is trained with only 40% data of the dataset using binary cross-entropy loss function and the intersection of union (IOU) loss function. In the experiment, the performance of the proposed method is evaluated using the rail surface defect dataset (RSDD) dataset by comparing it with other methods. The experimental results show that the segmentation performance and real-time performance of the proposed method are better than those of other methods.

Similar Papers
  • PDF
  • Research Article
  • Citations19

Swin Transformer Combined with Convolution Neural Network for Surface Defect Detection

  • Nov 16, 2022
  • Machines
  • Yinghao Li +4
  • Conference Article
  • Citations88

Detection of rail surface defects based on CNN image recognition and classification

  • Feb 01, 2018
  • Lidan Shang +4
  • Research Article
  • Citations95

Defect detection method for rail surface based on line-structured light

  • Mar 23, 2020
  • Measurement
  • Xiaohui Cao +3
  • Research Article
  • Citations9

A lightweight convolutional neural network for surface defect detection in strip steel

  • Mar 11, 2025
  • Measurement Science and Technology
  • Chunlong Yang +5
  • Research Article
  • Citations3

Variational Autoencoder Diffusion Model (VAEDM) and Divergence Asynchronous Reinforcement Learning (DARL) for Rail Surface Defect Detection

  • May 30, 2024
  • Journal of Internet Services and Information Security
  • Samyuktha Sasi Sekaran +1
  • Research Article
  • Citations19

Lightweight and attention-based CNN architecture for wildfire detection using UAV vision data

  • Sep 17, 2023
  • International Journal of Remote Sensing
  • Rahmi Arda Aral +2
  • Research Article

YOLO-GSD-seg: YOLO for Guide Rail Surface Defect Segmentation and Detection

  • Jan 26, 2026
  • Applied Sciences
  • S Lai +4
  • Research Article
  • Citations54

CUFuse: Camera and Ultrasound Data Fusion for Rail Defect Detection

  • Nov 01, 2022
  • IEEE Transactions on Intelligent Transportation Systems
  • Zhengxing Chen +5
  • Research Article
  • Citations24

ELCNN: A Deep Neural Network for Small Object Defect Detection of Magnetic Tile

  • Jan 01, 2022
  • IEEE Transactions on Instrumentation and Measurement
  • Weijian Liang +1
  • Research Article
  • Citations6

Automated peanut defect detection using hyperspectral imaging and deep learning: A real-time approach for smart agriculture

  • Aug 01, 2025
  • Smart Agricultural Technology
  • Shih-Yu Chen +8
  • Conference Article
  • Citations3

Surface defects detection algorithm for a small sample of sealing rings for aerospace based on deep learning

  • Jan 31, 2020
  • 刘 辉 +3
  • Research Article

A UAV-Based Framework for Visual Detection and Geospatial Mapping of Real Road Surface Defects

  • Feb 07, 2026
  • Drones
  • Paula López +3
  • Research Article
  • Citations6

Electrolytic capacitor surface defect detection based on deep convolution neural network

  • Jan 18, 2024
  • Journal of King Saud University - Computer and Information Sciences
  • Haijian Wang +3
  • Research Article
  • Citations3

Lamb mode identification based on lightweight CNN

  • Oct 12, 2024
  • Nondestructive Testing and Evaluation
  • Juanjuan Li +1
  • Research Article
  • Citations11

A Lightweight Convolutional Spiking Neural Network for Fires Detection Based on Acoustics

  • Jul 26, 2024
  • Electronics
  • Xiaohuan Li +3
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.