Recognition of wheat rusts in a field environment based on improved DenseNet

Abstract

Currently, the main methods for detecting plant diseases are sampling and manual visual inspection. However, these methods are time-consuming, laborious and prone to misinterpretation. Rapid advances in Deep Learning (DL) techniques offer new possibilities. This study focused on analysing the confounding factors among three types of wheat rust (stripe rust, leaf rust and stem rust) and aimed to achieve higher classification accuracy. The following approaches were used: (1) Images were collected from several crops and diseases: Wheat Rusts Dataset (WRD), Wheat Common Disease Dataset (WDD), and Common Poaceae Disease Dataset (PDD); (2) Seven common convolutional neural network (CNN) models were made and their performance compared. DenseNet121 was selected as the base model, and its classification results further analysed. The results of the above analyses were then considered using phenotypic morphology and model structure analysis, as well as potential confounder discussions; (3) Adjustments and optimisations were made based on the identified confounding factors. The final improved model, designated Imp-DenseNet, achieved the following accuracies with different datasets: Top-1 accuracy = 98.32% (WRD), Top-3 accuracy = 97.30% (WDD) and Top-5 accuracy = 96.60% (PDD) (Top-x Accuracy refers to the accuracy of the top-ranked category that matches or containing the actual results). The study revealed the potential factors contributing to the confusion among the three wheat rusts and successfully achieved higher accuracy. It can provide a new perspective for future research on other diseases of wheat or other crops.