Research Repository

Wheat stripe rust grading by deep learning with attention mechanism and images from mobile devices

Mi, Zhiwen and Zhang, Xudong and Su, Jinya and Han, Dejun and Su, Baofeng (2020) 'Wheat stripe rust grading by deep learning with attention mechanism and images from mobile devices.' Frontiers in Plant Science, 11. 558126-. ISSN 1664-462X

fpls-11-558126.pdf - Published Version
Available under License Creative Commons Attribution.

Download (3MB) | Preview


Wheat stripe rust is one of the main wheat diseases worldwide, which has significantly adverse effects on wheat yield and quality, posing serious threats on food security. Disease severity grading plays a paramount role in stripe rust disease management including breeding disease-resistant wheat varieties. Manual inspection is time-consuming, labor-intensive and prone to human errors, therefore, there is a clearly urgent need to develop more effective and efficient disease grading strategy by using automated approaches. However, the differences between wheat leaves of different levels of stripe rust infection are usually tiny and subtle, and, as a result, ordinary deep learning networks fail to achieve satisfying performance. By formulating this challenge as a fine-grained image classification problem, this study proposes a novel deep learning network C-DenseNet which embeds Convolutional Block Attention Module (CBAM) in the densely connected convolutional network (DenseNet). The performance of C-DenseNet and its variants is demonstrated via a newly collected wheat stripe rust grading dataset (WSRgrading dataset) at Northwest A&F University, Shaanxi Province, China, which contains a total of 5,242 wheat leaf images with 6 levels of stripe rust infection. The dataset was collected by using various mobile devices in the natural field condition. Comparative experiments show that C-DenseNet with a test accuracy of 97.99% outperforms the classical DenseNet (92.53%) and ResNet (73.43%). GradCAM++ network visualization also shows that C-DenseNet is able to pay more attention to the key areas in making the decision. It is concluded that C-DenseNet with an attention mechanism is suitable for wheat stripe rust disease grading in field conditions.

Item Type: Article
Uncontrolled Keywords: Wheat stripe rust; Disease grading; mobile images; image processing; precision agriculture
Divisions: Faculty of Science and Health
Faculty of Science and Health > Computer Science and Electronic Engineering, School of
SWORD Depositor: Elements
Depositing User: Elements
Date Deposited: 09 Sep 2020 12:50
Last Modified: 18 Aug 2022 13:00

Actions (login required)

View Item View Item