ADHI-YOLO: Attentive SPPF with Dynamic Head Integration for Object Detection in Low-Light Environments
摘要
Object detection in low-light environments presents significant challenges, as existing algorithms often struggle under such conditions. To address this, we propose ADHI-YOLO, a novel method optimized for low-light images, improving both detection accuracy and robustness. ADHI-YOLO integrates the large-kernel Attention Feature Pyramid Module (AFAM), an improved Dynamic Attention Head Module (DPH), and the innovative I-MPDIOU loss function, which effectively tackles image noise and detail loss. Additionally, mixed-precision training enhances efficiency. Experimental results on the ExDark dataset show that ADHI-YOLO achieves a mean Average Precision (mAP) of 74.9%, outperforming existing algorithms, making it a promising solution for low-light object detection.