MTS-YOLO: A Multi-Task Lightweight and Efficient Model for Tomato Fruit Bunch Maturity and Stem Detection
2024
Maonian Wu | Hanran Lin | Xingren Shi | Shaojun Zhu | Bo Zheng
The accurate identification of tomato maturity and picking positions is essential for efficient picking. Current deep-learning models face challenges such as large parameter sizes, single-task limitations, and insufficient precision. This study proposes MTS-YOLO, a lightweight and efficient model for detecting tomato fruit bunch maturity and stem picking positions. We reconstruct the YOLOv8 neck network and propose the high- and low-level interactive screening path aggregation network (HLIS-PAN), which achieves excellent multi-scale feature extraction through the alternating screening and fusion of high- and low-level information while reducing the number of parameters. Furthermore, We utilize DySample for efficient upsampling, bypassing complex kernel computations with point sampling. Moreover, context anchor attention (CAA) is introduced to enhance the model’s ability to recognize elongated targets such as tomato fruit bunches and stems. Experimental results indicate that MTS-YOLO achieves an F1-score of 88.7% and an [email protected] of 92.0%. Compared to mainstream models, MTS-YOLO not only enhances accuracy but also optimizes the model size, effectively reducing computational costs and inference time. The model precisely identifies the foreground targets that need to be harvested while ignoring background objects, contributing to improved picking efficiency. This study provides a lightweight and efficient technical solution for intelligent agricultural picking.
显示更多 [+] 显示较少 [-]