Real-Time Traffic Load Monitoring Framework Based on Deep Learning Model and Statistical Regularities of Vehicle Shape Prior Information
摘要
Existing vision-based traffic load monitoring systems only provide vehicle bounding boxes, which makes it challenging to accurately determine the load positions. This paper proposes a framework to determine real-time traffic load distribution. The proposed framework uses the YOLO-v5 model to identify vehicles on the monitoring screen and makes reasonable assumptions about the geometric shape of the vehicles. These assumptions are used as prior information to propose an indirect calculation method for vehicle equivalent concentrated load, and a load position coordinates dataset is labeled. Finally, the statistical law of the labeled data is analyzed to verify its normality, and the maximum likelihood estimation of the load coordinate is determined. The proposed method is tested in the field, combined with a Kalman filter to track vehicles and optimize the load trajectory. The results show that the proposed method significantly improves the accuracy of the original algorithm and has strong practical application value.