计算机科学
对象(语法)
计算机视觉
目标检测
人工智能
算法
模式识别(心理学)
作者
Jian Li,Xin Wang,Qi Chang,Yongshan Wang,Haifeng Chen
出处
期刊:Electronics
[Multidisciplinary Digital Publishing Institute]
日期:2024-09-05
卷期号:13 (17): 3527-3527
被引量:4
标识
DOI:10.3390/electronics13173527
摘要
In low-light environments, the presence of numerous small, dense, and occluded objects challenges the effectiveness of conventional object detection methods, failing to achieve desirable results. To address this, this paper proposes an efficient object detection network, YOLO_GD, which is designed for precise detection of targets in low-light scenarios. This algorithm, based on the foundational framework of YOLOv5s, implements a cross-layer feature fusion method founded on an information gathering and distribution mechanism. This method mitigates the issue of information loss during inter-layer feature exchange and, building on this, constructs a Bi-level routing spatial attention module to reduce computational redundancy caused by the self-attention mechanism, thereby enhancing the model’s detection accuracy for small objects. Furthermore, through the introduction of a novel deformable convolution, a cross-stage local feature fusion module is established, enabling the model to capture the complex features of input data more accurately and improve detection precision for dense objects. Lastly, the introduction of a probabilistic distance metric in the bounding box regression loss function enhances the network model’s generalization capability, further increasing detection accuracy in occluded scenarios. Experimental results on the ExDark dataset demonstrate that compared to YOLOv5, there is a 5.97% improvement in mean average precision (mAP), effectively enhancing object detection performance in low-light conditions.
科研通智能强力驱动
Strongly Powered by AbleSci AI