- Conference Article
- 10.1109/iceace67491.2025.11439392
Research on the Dual-Task Algorithm for Vehicle Detection and Traffic Prediction in Complex Weather Based on Collaborative Optimization of YOLOv11 and Logistic Regression
- Dec 26, 2025
- Zhang Yichen
With the rapid development of intelligent transportation systems, vehicle detection and traffic flow statistics have become increasingly critical in traffic management and the construction of smart cities. However, complex weather conditions lead to challenges in vehicle detection algorithms, increasing false negatives and false positives. Additionally, the rapid growth of vehicle ownership makes traditional algorithms inadequate for meeting practical application needs. Therefore, this paper proposes a collaborative optimization-based method combining YOLOv11 for vehicle detection and traffic flow analysis under complex weather conditions. Vehicle information is accurately extracted through object detection and tracking methods, and further analyzed using Logistic Regression to assess traffic congestion. The main contributions of this study are: (1) Integration of the CBAM attention mechanism into YOLOv11, enhancing the vehicle feature extraction ability through collaborative channel and spatial attention, thereby improving interference resistance under complex weather conditions. (2) Adoption of the Shape-IoU loss function, with dynamic weight allocation to optimize the accuracy of detecting smaller vehicles and the precision of bounding box regression in images where vehicles are located at a greater distance. (3) Construction of a traffic congestion prediction model based on the detection results, integrated with Logistic Regression. The proposed method's effectiveness is validated on the public dataset UA-DETRAC. Through the synergistic effect of the CBAM attention mechanism and Shape-IoU loss function, the improved model's mAP50-95 shows a 0.3 percentage point increase over the YOLOv11 baseline, providing efficient and reliable technical support for intelligent transportation systems. This study enhances the prediction performance of traffic states under complex weather conditions by effectively calculating traffic flow parameters such as vehicle count and density, thus offering foundational data support for traffic management decision-making. The research conclusions are of practical significance for urban traffic improvement.
Read more