Automatic Counting of Lettuce Using an Improved YOLOv5s with Multiple Lightweight Strategies

Pan Zhang,Daoliang Li
DOI: https://doi.org/10.1016/j.eswa.2023.120220
IF: 8.5
2023-01-01
Expert Systems with Applications
Abstract:Crop counting plays different roles in different growth stages. The counting of the early stage is used to evaluate seed germination rate and seedling survival rate, the counting during growth is used to evaluate plant mortality caused by potential pests and diseasesand, and the counting of the later stage is used to accurately evaluate crop yield. Therefore, how to realize the automatic, fast, and accurate counting of crops during the whole growth period is a key problem to be solved urgently. The development of computer vision and artificial intelligence technology has laid good technical support for crop counting. According to the growth characteristics and cultivation process of lettuce, combined with the requirements of model generalization performance, this study collected image datasets of 9 growth stages of multiple varieties of lettuce. Then, according to the specific requirements of model deployment in the actual production scenario, taking YOLOv5s as the benchmark model, a variety of lightweight strategies are used to improve it, to find the best balance between model performance, size, and inference speed. Finally, this study proposes a lightweight YOLOv5s lettuce counting model based on ShuffleNetv2. The mAP of the model is 99.42%, Recall is 99.13%, Precision is 98.24%, F1-score is 0.99, the model-size is only 3.18 M, and the inference time is only 1.0 ms. Compared with other methods, this model has significant advantages in real-time, fast, and accurate counting of lettuce. In the actual planting scenes with small, big, and crowded targets, the measured plant values and predicted plant values of the model are highly consistent.
What problem does this paper attempt to address?