Drone model classification using convolutional neural network trained on synthetic data
dc.contributor.author | Wisniewski, Mariusz | |
dc.contributor.author | Rana, Zeeshan A. | |
dc.contributor.author | Petrunin, Ivan | |
dc.date.accessioned | 2022-08-23T12:34:59Z | |
dc.date.available | 2022-08-23T12:34:59Z | |
dc.date.issued | 2022-08-12 | |
dc.description.abstract | We present a convolutional neural network (CNN) that identifies drone models in real-life videos. The neural network is trained on synthetic images and tested on a real-life dataset of drone videos. To create the training and validation datasets, we show a method of generating synthetic drone images. Domain randomization is used to vary the simulation parameters such as model textures, background images, and orientation. Three common drone models are classified: DJI Phantom, DJI Mavic, and DJI Inspire. To test the performance of the neural network model, Anti-UAV, a real-life dataset of flying drones is used. The proposed method reduces the time-cost associated with manually labelling drones, and we prove that it is transferable to real-life videos. The CNN achieves an overall accuracy of 92.4%, a precision of 88.8%, a recall of 88.6%, and an f1 score of 88.7% when tested on the real-life dataset. | en_UK |
dc.identifier.citation | Wisniewski M, Rana ZA, Petrunin I. (2022) Drone model classification using convolutional neural network trained on synthetic data. Journal of Imaging, Volume 8, Issue 8, August 2022, Article number 218 | en_UK |
dc.identifier.issn | 2313-433X | |
dc.identifier.uri | https://doi.org/10.3390/jimaging8080218 | |
dc.identifier.uri | https://dspace.lib.cranfield.ac.uk/handle/1826/18350 | |
dc.language.iso | en | en_UK |
dc.publisher | MDPI | en_UK |
dc.relation.isreferencedby | https://dspace.lib.cranfield.ac.uk/handle/1826/23082 | |
dc.relation.isreferencedby | https://doi.org/10.17862/cranfield.rd.19423925 | |
dc.rights | Attribution 4.0 International | * |
dc.rights.uri | http://creativecommons.org/licenses/by/4.0/ | * |
dc.subject | unmanned aerial vehicles | en_UK |
dc.subject | drones | en_UK |
dc.subject | airport security | en_UK |
dc.subject | convolutional neural network | en_UK |
dc.subject | synthetic images | en_UK |
dc.subject | synthetic data | en_UK |
dc.subject | domain randomization | en_UK |
dc.subject | drone detection | en_UK |
dc.subject | drone classification | en_UK |
dc.subject | drone identification | en_UK |
dc.subject | artificial intelligence | en_UK |
dc.title | Drone model classification using convolutional neural network trained on synthetic data | en_UK |
dc.type | Article | en_UK |