Enhancing pedestrian detection in adverse conditions using YOLOv5s with adaptive weighted fusion efficient channel attention module
TL;DR
An enhanced pedestrian detection framework based on you only look one version 5 (YOLOv5s), retrained on an augmented common object in context (COCO) dataset focused on the person class is introduced, demonstrating the efficacy of the proposed attention module in improving detection robustness under challenging environmental conditions.
Abstract
Pedestrian detection constitutes a critical task within advanced driver assistance systems (ADAS), where reliable identification of pedestrians is essential for ensuring vehicular safety. Although deep learning has substantially improved detection performance, existing state-of-the-art models continue to exhibit notable degradation in adverse weather conditions and low-light. To mitigate these challenges, this study introduces an enhanced pedestrian detection framework based on you only look one version 5 (YOLOv5s), retrained on an augmented common object in context (COCO) dataset focused on the person class. Additionally, a novel, lightweight, and adaptive attention mechanism called: the weighted fusion efficient channel attention (WF-ECA) module is incorporated into the detection architecture. The WF-ECA module selectively focusses on important features without compromising computational efficiency or inference speed. Comparative experiments demonstrate a 5% increase in mean average precision (mAP) in comparison to the baseline model, thereby demonstrating the efficacy of the proposed attention module in improving detection robustness under challenging environmental conditions. These findings highlight the potential of attention-based mechanisms to enhance pedestrian detection performance in real-world ADAS applications.