Home > Published Issues > 2023 > Volume 11, No. 3, September 2023 >
JOIG 2023 Vol.11(3): 263-270
doi: 10.18178/joig.11.3.263-270

Systematic Configuration for Hyperparameters Optimization in Transferring of CNN Model to Disaster Events Classification from UAV Images

Supaporn Bunrit *, V Nittaya Kerdprasop, and Kittisak Kerdprasop
School of Computer Engineering, Institute of Engineering, Suranaree University of Technology, Nakhon Ratchasima, Thailand; Email: nittaya@sut.ac.th (N.K.), kerdpras@sut.ac.th (K.K.)
*Correspondence: sbunrit@sut.ac.th (S.B.)

Manuscript received March 31, 2023; revised April 20, 2023; accepted May 30, 2023.

Abstract—Deep learning and computer vision-based approaches incorporated with the evolution of the relevant technologies of Unmanned Aerial Vehicles (UAVs) and drones have significantly motivated the advancements of disaster management applications. This research studied a classification method for disaster event identification from UAV images that is suitable for disaster monitoring. A Convolution Neural Network (CNN) of GoogleNet models that were pretrained from ImageNet and Place365 datasets was explored to find the appropriate one for fine-tuning to classify the disaster events. In order to get the optimal performance, a systematic configuration for searching the hyperparameters in fine-tuning the CNN model was proposed. The top three hyperparameters that affect the performance, which are the initial learning rate, the number of epochs, and the minibatch size, were systematically set and tuned for each configuration. The proposed approach consists of five stages, during which three types of trials were used to monitor different sets of the hyperparameters. The experimental result revealed that by applying the proposed approach the model performance can increase up to 5%. The optimal performance achieved was 98.77 percent accuracy. For UAV/drone applications, where a small onboard model is preferred, GoogleNet that is quite small in model size and has a good structure for further fine tuning is suitable to deploy.

Keywords—hyperparameter optimization, transfer learning, fine-tuning, convolution neural network, deep learning, disaster events classification, Unmanned Aerial Vehicles (UAVs), Drones

Cite: Supaporn Bunrit, Nittaya Kerdprasop, and Kittisak Kerdprasop, "Systematic Configuration for Hyperparameters Optimization in Transferring of CNN Model to Disaster Events Classification from UAV Images," Journal of Image and Graphics, Vol. 11, No. 3, pp. 263-270, September 2023.

Copyright © 2023 by the authors. This is an open access article distributed under the Creative Commons Attribution License (CC BY-NC-ND 4.0), which permits use, distribution and reproduction in any medium, provided that the article is properly cited, the use is non-commercial and no modifications or adaptations are made.