GPC-YOLO: An Improved Lightweight YOLOv8n Network for the Detection of Tomato Maturity in Unstructured Natural...

Effective fruit identification and maturity detection are important for harvesting and managing tomatoes. Current deep learning detection algorithms typically demand significant computational resources and memory. Detecting severely stacked and obscured tomatoes in unstructured natural environments is challenging because of target stacking, target occlusion, natural illumination, and background noise. The proposed method involves a new lightweight model called GPC-YOLO based on YOLOv8n for tomato identification and maturity detection. This study proposes a C2f-PC module based on partial convolution (PConv) for less computation, which replaced the original C2f feature extraction module of YOLOv8n. The regular convolution was replaced with the lightweight Grouped Spatial Convolution (GSConv) by downsampling to reduce the computational burden. The neck network was replaced with the convolutional neural network-based cross-scale feature fusion (CCFF) module to enhance the adaptability of the model to scale changes and to detect many small-scaled objects. Additionally, the integration of the simple attention mechanism (SimAM) and efficient intersection over union (EIoU) loss were implemented to further enhance the detection accuracy by leveraging these lightweight improvements. The GPC-YOLO model was trained and validated on a dataset of 1249 mobile phone images of tomatoes. Compared to the original YOLOv8n, GPC-YOLO achieved high-performance metrics, e.g., reducing the parameter number to 1.2 M (by 59.9%), compressing the model size to 2.7 M (by 57.1%), decreasing the floating point of operations to 4.5 G (by 45.1%), and improving the accuracy to 98.7% (by 0.3%), with a detection speed of 201 FPS. This study showed that GPC-YOLO could effectively identify tomato fruit and detect fruit maturity in unstructured natural environments. The model has immense potential for tomato ripeness detection and automated picking applications.