DBA-YOLO: a mamba-incorporated dual-branch axial feature enhancement network for small object detection in traffic scenarios
摘要
This paper presents DBA-YOLO, a dual-branch axial feature enhancement network based on YOLOv12, aimed at small object detection in complex traffic scenarios. The model addresses misdetection and missed detection of small objects due to lighting variations and occlusions. Key contributions include: 1) the MLAA (Mamba-Like Axial Attention) module, which integrates a Mamba-style gating mechanism and achieves accurate small object detection through spatial-channel and segmented axial attention; 2) the HLGAB (Hierarchical Local-Global Attention Block) module, which enhances feature interaction via dual-path channels, reducing misdetection in dense scenes. Experiments on the BDD100K dataset show a