Lightweight CNN model: Automated vehicle detection in aerial images
Document Type
Article
Publication Date
6-1-2023
Abstract
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.
Keywords
Computer vision, Aerial imagery, Deep neural network, Object detection, Image
Divisions
sch_ecs
Funders
Industry-Driven Innovation Grant (IDIG) by Universiti Malaya [PPSI-2020-CLUSTER-SD01]
Publication Title
SIGNAL IMAGE AND VIDEO PROCESSING
Volume
17
Issue
4
Publisher
SPRINGER LONDON LTD
Publisher Location
236 GRAYS INN RD, 6TH FLOOR, LONDON WC1X 8HL, ENGLAND