Tomato pose estimation using the association of tomato body and sepal

Abstract

In facility horticulture smart farms, harvesting robotic systems to automate harvesting tasks are challenging due to the complex environment, irregular growth and fruits pose. To harvest a fruits, the vision information required by a harvesting robot is accurate position and pose. Especially, fruit pose information is essential for planning paths to avoid damaging stems, leaves, branches, and obstacles, preventing damage to the fruit and harvesting robots, and planning efficient harvest sequences. This paper presents a method that uses information from tomato sepals to estimate the orientation of tomatoes, which are a commonly grown crop in horticulture. First, we train a YOLOv8 model to detect and segment bodies and sepals of tomatoes. For robust training of the model, the training data is constructed using effective data augmentation, which synthesizes segmented foreground objects and inserting them into the background. Then, for finding association between the body and sepal of a tomato, we apply IoU score based matching and the Hungarian algorithm. Consequently, we obtain point clouds for both parts using RGB-D data. Finally, we compute the center point of each object by using spherical fitting and the statistics of the point cloud, respectively. Then, we estimate the pose of the tomato as a vector between two center points of the body and sepal. To accurately and practically evaluate the proposed method, we generated ground truth data using calibration patterns in tomato greenhouse. As the experimental results show, The segmentation results show that AP50,sepal is 94.7, AP50,tomato is 96.3, and mAP is 61.5. The result of the pose estimation for the pose validation dataset is a mean error angle of 6.79 ± 3.18 and angle errors of less than 10 degrees account for 87.2%. The total algorithm proposed in this paper requires about 0.038s for each tomato greenhouse image. As a result, our proposed pose estimation can be practically utilized in robot systems for tomato harvesting.