Object detection of railway tool based on multi-scale feature fusion network
DOI:
Author:
Affiliation:

1. Hubei Province Key Laboratory of Intelligent Robot, Wuhan Institute of Technology, Wuhan430205, China; 2. School of Mathematics and Physics, Wuhan Institute of Technology, Wuhan 430205, China

Clc Number:

TP391.4

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Image object detection is crucial for the automatic counting of railway tools. However, collected images of railway tools have the characteristics of low illumination, huge difference in the scale of different objects and complex backgrounds. Existing image object detection methods cannot detect railway tools efficiently. In this paper, we propose a novel object detection model which is able to enhance the object-detection ability by fusing multi-scale features. Based on the object detection deep learning model Retinanet, we construct a feature fusion enhancement module. By fusing features, our model can efficiently detect railway tools with different scales. Experiments were conducted based on real-world datasets. Results show that our method is more efficient than Retinanet, and the mAP is increased from 97.85% to 98.11%. By the accurate detection of railway tools in complex backgrounds, our approach can lay the technical foundation for intelligent railway operation and maintenance.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: April 02,2024
  • Published: