Advanced Machine Learning Method for Watermelon Identification and Yield Estimation

Abstract

Watermelon is a popular fruit, predominantly cultivated in Asian countries. However, the production and harvesting processes present several challenges. Due to its size and weight, manually harvesting watermelons is labor-intensive and costly. In the future, technology is expected to enable robots to harvest watermelons. Therefore, it becomes essential to introduce intelligent systems to effectively identify and locate watermelons in harvesting. This research aims to develop an advanced methodology for watermelon identification and location using You Look Only Once (YOLO)v8 and YOLOv8-oriented bounding box (OBB) algorithms. Furthermore, the simple online and real-time tracking (SORT) algorithm was employed to track and count watermelons and estimate yield. The performance of YOLOv8-OBB was better than that of YOLOv8 and the highest precision (0.938) was achieved by YOLOv8s-OBB. Additionally, the size of each watermelon was measured with both models. The models help farmers find the optimal watermelons for harvest.