Agriculture (Feb 2024)

DiffuCNN: Tobacco Disease Identification and Grading Model in Low-Resolution Complex Agricultural Scenes

  • Huizhong Xiong,
  • Xiaotong Gao,
  • Ningyi Zhang,
  • Haoxiong He,
  • Weidong Tang,
  • Yingqiu Yang,
  • Yuqian Chen,
  • Yang Jiao,
  • Yihong Song,
  • Shuo Yan

DOI
https://doi.org/10.3390/agriculture14020318
Journal volume & issue
Vol. 14, no. 2
p. 318

Abstract

Read online

A novel deep learning model, DiffuCNN, is introduced in this paper, specifically designed for counting tobacco lesions in complex agricultural settings. By integrating advanced image processing techniques with deep learning methodologies, the model significantly enhances the accuracy of detecting tobacco lesions under low-resolution conditions. After detecting lesions, the grading of the disease severity is achieved through counting. The key features of DiffuCNN include a resolution enhancement module based on diffusion, an object detection network optimized through filter pruning, and the employment of the CentralSGD optimization algorithm. Experimental results demonstrate that DiffuCNN surpasses other models in precision, with respective values of 0.98 on precision, 0.96 on recall, 0.97 on accuracy, and 62 FPS. Particularly in counting tobacco lesions, DiffuCNN exhibits an exceptional performance, attributable to its efficient network architecture and advanced image processing techniques. The resolution enhancement module based on diffusion amplifies minute details and features in images, enabling the model to more effectively recognize and count tobacco lesions. Concurrently, filter pruning technology reduces the model’s parameter count and computational burden, enhancing the processing speed while retaining the capability to recognize key features. The application of the CentralSGD optimization algorithm further improves the model’s training efficiency and final performance. Moreover, an ablation study meticulously analyzes the contribution of each component within DiffuCNN. The results reveal that each component plays a crucial role in enhancing the model performance. The inclusion of the diffusion module significantly boosts the model’s precision and recall, highlighting the importance of optimizing at the model’s input end. The use of filter pruning and the CentralSGD optimization algorithm effectively elevates the model’s computational efficiency and detection accuracy.

Keywords