An Efficient Detector for Automatic Tomato Classification Systems

Abstract

Nowadays, artificial intelligence and robotics have been deployed in almost all areas of human life. Especially in agriculture, it has helped people free up labor, speed up production, and ensure product quality. This research aims to develop a vision-based tomato detector to support robots and automatic tomato classification systems. In this idea, the original convolution blocks in the Backbone and Neck modules of the YOLOv8n architecture are replaced by a new version, called Receptive Field Attention Convolution (RFAConv). The model was trained and evaluated using four benchmarks, including the Laboro Tomato, Tomato Plantfactory, Sai Gon University, and CubeAI datasets. It achieved the best performance at 89.8% of [email protected] and 65.2% of [email protected]:0.95 on the Tomato Plantfactory dataset. These results show that the proposed network is superior to others under a fixed set of experimental conditions. The video demonstration, datasets, weights, and models are available at https://bit.ly/3XdLrLM.

Keywords

• Convolutional neural network (CNN)
• tomato detection and classification
• receptive field attention convolution (RFAConv)
• YOLOv8n