Lightweight CNN model: Automated vehicle detection in aerial images

• Main Authors: Momin, Md Abdul, Junos, Mohamad Haniff, Khairuddin, Anis Salwa Mohd, Abu Talip, Mohamad Sofian
• Format: Article
• Published: SPRINGER LONDON LTD 2023
• Subjects: TK Electrical engineering. Electronics Nuclear engineering; TR Photography
• Online Access: http://eprints.um.edu.my/39488/

Efficient vehicle detection has played an important role in Intelligent Transportation Systems (ITS) in smart cities. With the development of the Convolutional Neural Network (CNN) for objection detection, new applications have been designed to enable on-road vehicle detection algorithms. Therefore, this work aims to further improve the conventional CNN model for real-time detection on low-cost embedded hardware. In this study, a lightweight CNN model is proposed based on YOLOv4 Tiny to detect vehicles from the VEDAI dataset. In the proposed method, one additional scale feature map is added to make a total of three prediction boxes in the architecture. Then, the output image size of the second and third prediction boxes are upscaled in order to improve detection accuracy in detecting small size vehicles in the aerial images. The proposed model has been evaluated on NVIDIA Geforce 940MX GPU-based computer, Google Collab (TESLA K80) and Jetson Nano. Based on the experimental results, this study has demonstrated that the proposed model achieved better mean average precision (mAP) compared to the conventional YOLOv4 Tiny and previous works.