Real UAV-Bird Image Classification Using CNN with a Synthetic Dataset
Autor: | Ergun Erçelebi, Ali Emre Öztürk |
---|---|
Jazyk: | angličtina |
Rok vydání: | 2021 |
Předmět: |
0209 industrial biotechnology
Technology Computer science QH301-705.5 QC1-999 Corner detection 02 engineering and technology synthetic image generation Synthetic data Image (mathematics) 020901 industrial engineering & automation 0202 electrical engineering electronic engineering information engineering General Materials Science Layer (object-oriented design) Biology (General) Instrumentation QD1-999 Dropout (neural networks) Fluid Flow and Transfer Processes Hyperparameter Contextual image classification business.industry Process Chemistry and Technology Deep learning Physics General Engineering deep learning Pattern recognition Engineering (General). Civil engineering (General) corner detection and nearest three-point selection (CDNTS) Computer Science Applications Chemistry corner detection rotary-wing UAV 020201 artificial intelligence & image processing Artificial intelligence TA1-2040 business CNN image classification |
Zdroj: | Applied Sciences Volume 11 Issue 9 Applied Sciences, Vol 11, Iss 3863, p 3863 (2021) |
ISSN: | 2076-3417 |
DOI: | 10.3390/app11093863 |
Popis: | A large amount of training image data is required for solving image classification problems using deep learning (DL) networks. In this study, we aimed to train DL networks with synthetic images generated by using a game engine and determine the effects of the networks on performance when solving real-image classification problems. The study presents the results of using corner detection and nearest three-point selection (CDNTS) layers to classify bird and rotary-wing unmanned aerial vehicle (RW-UAV) images, provides a comprehensive comparison of two different experimental setups, and emphasizes the significant improvements in the performance in deep learning-based networks due to the inclusion of a CDNTS layer. Experiment 1 corresponds to training the commonly used deep learning-based networks with synthetic data and an image classification test on real data. Experiment 2 corresponds to training the CDNTS layer and commonly used deep learning-based networks with synthetic data and an image classification test on real data. In experiment 1, the best area under the curve (AUC) value for the image classification test accuracy was measured as 72%. In experiment 2, using the CDNTS layer, the AUC value for the image classification test accuracy was measured as 88.9%. A total of 432 different combinations of trainings were investigated in the experimental setups. The experiments were trained with various DL networks using four different optimizers by considering all combinations of batch size, learning rate, and dropout hyperparameters. The test accuracy AUC values for networks in experiment 1 ranged from 55% to 74%, whereas the test accuracy AUC values in experiment 2 networks with a CDNTS layer ranged from 76% to 89.9%. It was observed that the CDNTS layer has considerable effects on the image classification accuracy performance of deep learning-based networks. AUC, F-score, and test accuracy measures were used to validate the success of the networks. |
Databáze: | OpenAIRE |
Externí odkaz: |