Embedded AI for Wheat Yellow Rust Infection Type Classification

Wheat is the most important and dominating crop in Pakistan in terms of production and acreage, which is grown on 37% of the cultivated area, accounting for 70% of the total production. However, wheat yield is highly affected by stripe rust, which is considered the most devastating fungal disease, causing 5.5 million tonnes of loss per year globally. In order to minimize this loss, the accurate and timely detection of rust disease is crucial instead of manual inspection. Towards this end, we propose a system to detect wheat rust disease and classify its infection types into four classes, including healthy, resistant, moderate (moderately resistant to moderately susceptible), and susceptible. The wheat rust dataset is collected indigenously from the National Agricultural Research Centre, Islamabad. A pre-trained U2 Net model is used to remove the background and extract the leaf containing the rust disease. Subsequently, two deep learning classifiers, including the Xception model and ResNet-50 are applied to classify the stripe rust severity levels, where the ResNet-50 model outperformed with the highest accuracy of 96%. This research presents a comparison between two state-of-the-art deep learning classifiers in terms of accuracy, memory utilization, and prediction time, which will assist the research community in selecting the most appropriate model for plant disease detection. Moreover, to assess the external validity, the performance of these classifiers is compared with the existing technique using a publicly available dataset, which confirms the validity of the results. Additionally, an intelligent edge computing rust detection device has been developed, where the trained ResNet-50 model is deployed, which facilitates the farmers to monitor the rust attack. The proposed research is aimed to assist the agricultural community to employ preventive measures in a site-specific manner based on the accurate diagnosis of rust disease & its severity, which is intended to improve the quality of the wheat as well as production.