UAV4TREE: DEEP LEARNING-BASED SYSTEM FOR AUTOMATIC CLASSIFICATION OF TREE SPECIES USING RGB OPTICAL IMAGES OBTAINED BY AN UNMANNED AERIAL VEHICLE

Abstract

Automated tree classification from unmanned aerial vehicle (UAV) images is a challenging task with several applications in forest management and conservation. In this study, we propose UAV4Tree a Deep Learning based system that automatically classifies RGB optical images obtained by the UAV. In particular, we explore the use of augmented datasets and various deep learning models, including ResNet, DenseNet, InceptionV3, and Vision Transformer, for the classification of tree images obtained from UAVs. Our experiments show that the use of an augmented dataset can significantly improve the accuracy of the classification by approximately 10 points compared to the use of a non-augmented dataset. We also found that fine-tuning and the introduction of dropout were essential for improving the generalization ability of the models on the augmented dataset. Furthermore, the use of Super Resolution Generative Adversarial Network (SR-GAN) in the original dataset allowed us to increase the performance of some models. Our findings provide valuable insights into the use of deep learning models for automated tree classification from UAV imagery, which has significant implications for sustainable forest management and conservation.