A MSA-YOLO Obstacle Detection Algorithm for Rail Transit in Foggy Weather
Jian Chen, Donghui Li, Weiqiang Qu, Zhiwei Wang · Applied Sciences · 2024
Obstacles on rail transit significantly compromise operational safety, particularly under dense fog conditions. To address missed and false detections in traditional rail transit detection methods, this paper proposes a multi-scale adaptive YOLO (MSA-YOLO) algorithm. The algorithm incorporates six filters: defog, white balance, gamma, contrast, tone, and sharpen, to remove fog and enhance image quality. However, determining the hyperparameters of these filters is challenging. We employ a multi-scale adaptive module to optimize filter hyperparameters, enhancing fog removal and image quality. Subsequently, YOLO is utilized to detect obstacles on rail transit tracks. The experimental results are encouraging, demonstrating the effectiveness of our proposed method in foggy scenarios.