Ecological Indicators (Sep 2024)
Intelligent classification of maize straw types from UAV remote sensing images using DenseNet201 deep transfer learning algorithm
Abstract
China has abundant straw resources, but challenges in utilization persist. Utilization rates need improvement, and environmental pollution from straw burning remains a significant issue. Accurate and intelligent remote sensing classification of straw types is crucial for enhancing straw utilization and preventing straw burning. This paper proposed a new approach for the intelligent classification of maize straw types, using the DenseNet201 deep transfer learning algorithm based on RGB images captured by Unmanned Aerial Vehicle (UAV). The sample labels dataset was established for maize straw types, utilizing DenseNet201 deep transfer learning algorithm to pre-train the sample set. This pre-training facilitated model transfer and parameter initialization. Subsequently, the second round of deep transfer learning was performed to construct the final maize straw type remote sensing classification models using DenseNet201 deep transfer learning algorithm. This model and results were subsequently compared with maize straw type classification by the ResNet50 and GoogLeNet deep transfer learning algorithms, as well as maize straw type classification using DenseNet201, ResNet50, and GoogLeNet deep learning algorithms. The results showed that the accuracy of the pre-trained maize straw type deep transfer learning remote sensing classification model surpassed that of the untrained maize straw type deep learning remote sensing classification model, resulting in an enhancement of accuracy by 8.59%, 7.38%, and 1.28%, respectively. The DenseNet201 deep transfer learning model for maize straw types exhibited the highest accuracy with the overall accuracy of 95.57%, and the kappa coefficient of 0.9410. Hence, the DenseNet201 deep transfer learning classification of maize straw types enabled the attainment of intelligent remote sensing recognition of maize straw types. The classification methodology, model, and results presented in this paper can serve as valuable technical references, offering essential information support for agricultural and environmental protection departments actively involved in the comprehensive utilization of straw resources and atmospheric environmental protection efforts.
