Real-time object detection method of pineapple ripeness based on improved YOLOv8
CSTR:
Author:
Affiliation:

1.College of Electronic and Information Engineering, Guangdong Ocean University, Zhanjiang 524088, China;2.Guangdong Intelligent Ocean Sensor Network and Equipment Engineering Technology Research Center, Zhanjiang 524088, China;3.College of Mathematics and Computer Science, Guangdong Ocean University, Zhanjiang 524088, China

Clc Number:

TP391

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • | |
  • Comments
    Abstract:

    A real-time object detection method of pineapple ripeness based on improved YOLOv8 was proposed to improve the accuracy of mechanical harvesting of pineapples in planting areas with different ripeness and ensure the quality of pineapples. The improved model replaced the common convolutions in the backbone and neck parts of the original YOLOv8 model with depth-wise separable convolutions (DSConv) to streamline parameters of model to solve the problems of small object size, dense quantity, and light occlusion of picked mechanical pineapple picking in natural environments. Convolutional block attention mechanism (CBAM) module was introduced before feature fusion to prioritize important features and improve the accuracy of object detection. The original loss function CIoU of YOLOv8 network was replaced with the EIoU loss function to accelerate the speed of network convergence. The results showed that the mean of average precision (PmA) of the improved model for detecting the pineapple ripeness was 97.33%. The PmA of improved model was 5.53, 7.91, 4.38, and 4.66 percentage points higher than that of Faster R-CNN, YOLOv4, YOLOv5 and YOLOv7, respectively. The number of parameters of the algorithm model was only 16.8×106 on the premise of ensuring the accuracy of detection. It is indicated that the improved model improves the accuracy and inference speed of recognizing pineapple ripeness, and has stronger robustness.

    Table 6 Results of ablation experiments
    Table 7 Test comparison of different network models
    Table 1 The 4th degree of ripeness of pineapple
    Table 2 The 3rd degree of ripeness of pineapple
    Table 5 Comparison of experimental results between the improved YOLOv8 model and the original model
    Fig.1 Pineapple images under different shooting conditions
    Fig.2 Improved YOLOv8 model
    Fig.3 Depthwise separable convolution
    Fig.4 CBAM attention process diagram
    Fig.5 Channel attention module structure diagram
    Fig.6 Spatial attention module structure diagram
    Fig.7 Training process diagram
    Fig.8 Comparison of convolutional substitution effect
    Fig.9 Identification effect of the improved pre- and post- detection network
    Table 3 Basic information of pineapple maturity dataset
    Table 4 Experimental environment configuration
    Reference
    Related
    Cited by
Get Citation

周涛,王骥,麦仁贵. Real-time object detection method of pineapple ripeness based on improved YOLOv8[J]. Jorunal of Huazhong Agricultural University,2024,43(5):10-20.

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:December 07,2023
  • Online: October 08,2024
Article QR Code