关键词: 交通安全, 深度学习, 计算机视觉, 目标检测, 模型轻量化, 两轮自行车

Abstract:

To address the challenges of significant scale variations, numerous environmental interferences, high real-time requirements, and the difficulty in achieving a good balance between detection accuracy and computational cost faced by existing bicycle-type vehicle detection models, this study proposes YOLO-DBG, a lightweight and efficient bicycle-type vehicle detection model based on computer vision. First, a novel dual-branch pooling & depthwise separable convolution bottleneck module is designed, which synchronously extracts global contour and local detail features of bicycle-type vehicles through a differentiated feature aggregation strategy, thereby enhancing multiscale feature extraction capabilities and reducing model computational costs by integrating depthwise separable convolution. Second, a weighted bidirectional feature pyramid network architecture is introduced in the neck network, which enhances the fusion of key vehicle features through bidirectional cross-scale connections and a dynamic weighting mechanism, and effectively reduces model computational costs through node pruning. In addition, ghost convolution is used as a downsampling operator, which considerably compresses the model while maintaining the feature expression ability. These three modules work together to construct an effective lightweight network architecture. Experiments demonstrate that the proposed model achieves a 0.2% increase in mean average precision while reducing parameters, giga floating point operations, and model size by 55.8%, 37.0%, and 53.1%, respectively. The proposed method achieves ideal lightweighting without compromising detection accuracy, offering a novel solution for real-time detection of bicycle-type vehicles.

Key words: traffic safety, deep learning, computer vision, target detection, model lightweighting, bicycle-type vehicles