Performance Comparison of Lightweight CNN Models for Drone Collision Avoidance Dataset

Collision Avoidance System (CAS) is important for drone safety. CAS consists of three steps e.g., obstacle sensing, collision prediction, and collision avoidance. Collision prediction enables drones to gain information, process information, and estimate whether the object has a risk of collision. Convolution Neural Network (CNN) is one of the methods that can be employed for collision prediction. However, CNN is a method that needs a large data in the training. Dario Pedro et al. provided a dataset called the CoLANet dataset that consists of VDOs of collision drones. Subsequently, they proposed a new algorithm called Neural Network Pipeline which has a Convolution Neural Network (CNN) part to extract the feature from a couple of images. CNN extracts images by using MobileNetV2 as a pre-trained model. They chose MobileNetV2 due to training performance from another dataset. This paper aims to assess the performance of lightweight CNN models using the CoLANet dataset. The models will be trained on the Keras library with parameters of fewer than ten million. The models will be validated by Confusion Matrix and Receiver Operating Characteristics. In conclusion, we examine which pre-trained CNN model has the best performance and suggest ongoing work.