Tomato Ripeness Detection Model Based on Improved RT-DETR Lightweight Model
Abstract
Accurate tomato ripeness detection is crucial for automated harvesting; however, complex greenhouse environments—characterized by dynamic light interference, foliage occlusion, and dense fruit overlapping—severely hinder detection performance and lead to frequent misdetections. This study aims to develop a high-precision, lightweight detection model that simultaneously addresses these three core challenges, thereby providing a technically deployable algorithmic foundation for resource-constrained agricultural edge devices. To this end, we propose CFD-DETR, a lightweight tomato ripeness detection model based on the RT-DETR architecture. The model incorporates a CAEfficientViT backbone for the lightweight extraction of multi-scale color and texture features. Furthermore, a Focused Efficient Additive Attention (FEAA) mechanism is integrated to capture fine-grained local ripening traits with minimal computational overhead. During feature reconstruction, a Deep Dynamic Upsampling (DwDySample) operator is utilized to preserve semantic integrity. Additionally, we designed the Wise-SIoU loss function, which dynamically penalizes low-quality samples to enhance boundary fitting and robustness against background noise. Experimental evaluations demonstrate that CFD-DETR achieves 90.2% mAP@0.5, outperforming the baseline model by 2.1 percentage points while significantly reducing the parameter count and computational complexity by 47.2% and 52.5%, respectively. Cross-dataset validation on the publicly available Laboro Tomato and RaUTD datasets confirms the model’s superior generalization capabilities. Overall, CFD-DETR provides a highly efficient and robust solution for real-time agricultural robotics.
Fields of Study & Subjects
Smart Agriculture and AI
Greenhouse Technology and Climate Control
Plant Surface Properties and Treatments
Publication details
- Journal / source
- Agronomy
- Publisher
- MDPI AG
- Volume
- 16
- Issue
- 9
- Pages
- 932–932
- Publication date
- 4 May 2026
- Language
- EN