An improved pistachio detection approach using YOLO-v8 Deep Learning Models

Abstract

Pistachios are an agricultural product widely used in the food industry. It is very important that pistachios are presented to the consumer in good quality on time. At the same time, whether the shells of pistachios are open or closed is an important criterion from a commercial industrial point of view. Pistachios with their shells open have a high unsaturated fat content, a high maturity level and an expensive market value. In this study, the open or closed status of pistachios was determined by using Artificial Intelligence-based deep learning models. For pistachio detection, 423 image data belonging to the Pesteh dataset were classified using models of the Yolov8 algorithm, which detects objects using convolutional neural networks. The data set is divided into 80% training, 10% validation and 10% testing. The performances of the models were evaluated with precision, recall, F1 and mAP score metrics. The highest test mAP value of the Yolov8 algorithm, which was run with image data consisting of pistachios, was obtained with the Yolov8-m model with 94.8%. The Yolov8-m model achieved a very successful result with 49.6 MB weight size, 11.0 ms inference time value and 0.33 hours training time value. In addition, the model's fast classification performance and small file size facilitate its applicability in the industrial field. The results show that the classification and detection of open and closed shell pistachios has been successfully carried out with Yolo models.