FCAODNet: a fast freight train image detection model based on embedded FCA

Abstract

The fault detection of freight train image has some problems, such as low detection accuracy and slow detection speed. Aiming at the problem of slow detection speed in the process of train image fault detection, a lightweight object detection model fast channel attention network (FCAODNet) is proposed in this study. FCAODNet consists of four modules, including feature extraction network (FEN), lightweight multi-scale feature fusion (LMFF), prediction across scales (PAS), and decoding modules. FEN extracts image features, LMFF fuses features, PAS predicts the location of the target object, and the decoding module obtains the final prediction result. FCAODNet's FEN adopts CSPDarknet53tiny. The designed LMFF is embedded with two FCA modules to improve the detection accuracy. Experiments on train datasets and public datasets show that FCAODNet outperforms other state-of-the-art models in detection speed and has good detection accuracy and robustness.