A Fast Vehicle Counting and Traffic Volume Estimation Method Based on Convolutional Neural Network

Autor: Henglong Yang, Youmei Zhang, Yu Zhang, Hailong Meng, Shuang Li, Xianglin Dai
Jazyk: angličtina
Rok vydání: 2021
Předmět:
Zdroj: IEEE Access, Vol 9, Pp 150522-150531 (2021)
Druh dokumentu: article
ISSN: 2169-3536
DOI: 10.1109/ACCESS.2021.3124675
Popis: Vehicle counting and traffic volume estimation on traffic videos has gained extensive attention from multimedia and computer vision communities. Recent vehicle counting and volume estimation methods, including detection based and time-spatial image (TSI) based methods have achieved significant improvements. However, how to balance the accuracy and speed is still a challenge to this task. In this paper, we design a fast and accurate vehicle counting and traffic volume estimation method. Firstly, traffic videos are converted to TSIs and we annotate the vehicle locations in TSIs manually. Then, we design a simple TSI density map estimation network which utilizes attention mechanism to strengthen the features in the traffic locations for vehicle counting. Finally, we use the parameters obtained from the vehicle counting network to further estimate the traffic volume. Experiments on UA-DETRAC dataset demonstrate that the vehicle counting network not only takes a balance between counting accuracy and speed, but also well estimates the traffic volume when the video data is insufficient.
Databáze: Directory of Open Access Journals