HUBEI AGRICULTURAL SCIENCES ›› 2024, Vol. 63 ›› Issue (8): 23-27.doi: 10.14088/j.cnki.issn0439-8114.2024.08.005

• Image and Graphic Recognition • Previous Articles     Next Articles

Citrus fruit recognition in natural environment based on improved YOLOv8

YU Sheng-xin1, WEI Ying-ying1, FANG Hui1, LI Min1, CHAI Xiu-juan2, ZENG Zhi-kang1, QIN Ze-lin1   

  1. 1. Agricultural Science and Technology Information Research Institute of GAAS, Nanning 530000, China;
    2. Agricultural Information Institute of CAAS, Beijing 100000, China
  • Received:2024-04-02 Online:2024-08-25 Published:2024-09-05

Abstract: In order to achieve precise and fast identification of citrus fruits, an improved YOLOv8 was proposed. Firstly, certain traditional convolutions in the YOLOv8 were replaced with ODConv full-dimensional dynamic convolutions to enhance the model’s adaptability in complex natural environments. Subsequently, the CIoU loss function of YOLOv8 was substituted with the MPDIoU loss function to address the degradation issue of the CIoU loss function in specific scenarios. Furthermore, the effectiveness of ODConv full-dimensional dynamic convolutions and MPDIoU loss function was verified through a series of ablation experiments. The average recognition accuracy (mAP) of the improved models, YOLOv8n, YOLOv8s, YOLOv8m, YOLOv8l, and YOLOv8x, was increased from 86.40%, 88.92%, 88.97%, 88.99%, 89.11% to 88.25%, 89.32%, 89.57%, 89.90%, 90.12%, respectively. Experimental results demonstrated that ODConv full-dimensional dynamic convolutions and MPDIoU loss function significantly enhanced the citrus fruit identification capability of the YOLOv8 in natural environments.

Key words: citrus fruit recognition, convolutional neural network, YOLOv8, ODConv full-dimensional dynamic convolution, MPDIoU loss function

CLC Number: