FF-DEIM: DEIM with Image Dehazing and Self-Supervised Pretraining for Catenary Support Component Detection

The catenary support component (CSC) is a key part of the electrified railway system, and its operational status directly affects railway operational safety. These components’ images are collected using inspection equipment and detected using computer vision techniques. However, catenary network inspection faces the following issues: (1) due to limitations in the equipment’s shooting angle and changes in viewing distance, the collected images contain multi-scale and multi-class problems, and (2) the railway environment is highly variable, and adverse weather conditions such as fog, rain, and low light affect the imaging devices, leading to degraded image quality. To address these issues, this paper proposes a novel detection framework, FF-DEIM, for detecting catenary support components. First, a dual-channel fusion network (DCFNet) is introduced, which significantly improves image quality by removing foreground interferences such as fog, raindrops, and dynamic blur. Second, a pretraining framework based on contrastive learning, mask image modeling with contrastive learning (MIMCL), is designed to enhance the model’s focus on key regions of the catenary network components, optimizing feature extraction capabilities and improving model convergence speed. Then, a feature-focusing pyramid network (FFPN) is proposed, which uses the focus feature module to fuse cross-level contextual features, enhancing the ability to capture local details and improving the model’s small object detection performance. Finally, a drone-based catenary network image dataset, including scenes with fog, rain, and low light, is constructed, and experiments validate the effectiveness of the proposed method.