Diagnosis of tomato pests and diseases based on lightweight CNN model

Li Sun,Kaibo Liang,Yuzhi Wang,Wang Zeng,Xinyue Niu,Longhao Jin
DOI: https://doi.org/10.1007/s00500-023-08631-w
IF: 3.732
2023-06-09
Soft Computing
Abstract:Tomato crop yield can be negatively impacted by various diseases and pests. The application of computer vision to diagnose tomato diseases and pests presents two significant challenges. Firstly, the collected dataset of tomato pests and diseases is often imbalanced, leading to poor model performance. Secondly, mainstream models struggle to balance the relationship between training efficiency and accuracy. In this paper, we propose a novel method for diagnosing tomato pests and diseases using a lightweight network. Specifically, we develop Squeeze and SE Net (SSNet), a new convolutional neural network (CNN) based on SqueezeNet and SENet. We evaluate the performance of SSNet through comparative experiments for diagnosing tomato pests and diseases. Additionally, we examine the impact of dataset balance, data volume, and hyperparameters on model performance. Our results demonstrate that SSNet achieves model accuracies of 98.80% and 98.39% for tomato pests and diseases, respectively, with only 0.398 M parameters. This approach provides a promising and lightweight alternative to existing models.
computer science, artificial intelligence, interdisciplinary applications
What problem does this paper attempt to address?