Citrus fruit recognition method based on the improved model of YOLOv5
CSTR:
Author:
Affiliation:

1.College of Electronic Engineering (College of Artificial Intelligence), South China Agricultural University,Guangzhou 510642,China;2.Division of Citrus Machinery,China Agriculture Research System,Guangzhou 510642,China;3.Guangdong Engineering Research Center for Monitoring Agricultural Information, Guangzhou 510642,China;4.Guangdong Artificial Intelligence and Digital Economy Laboratory (Guangzhou), Guangzhou 510330,China;5.Engineering Fundamental Teaching and Training Center,South China Agricultural University, Guangzhou 510642,China

Clc Number:

S661.1;TP391.41

Fund Project:

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

    The rapid and accurate identification of citrus fruit is of great significance for the realization of automatically picking citrus in orchards,the prediction of citrus yield and the intelligent management of citrus industry. A citrus recognition method based on the improved model of YOLOv5 was proposed to realize the recognition of citrus fruits in natural environment.The feature extraction ability of the network and the problem of missed detection of occluded targets and small targets was improved by introducing the CBAM attention mechanism module. The α-IoU loss function instead of the GIoU loss function was used as the bounding box regression loss function to improve the positioning accuracy of the bounding box. The results showed that the average accuracy AP value of the proposed model reached 91.3%,with the detection time of a single citrus fruit image on the GPU of 16.7 ms and the model occupying 14.5 Mb of memory. It is indicated that the algorithm improved can quickly and accurately identify citrus fruits in the natural environment,meeting the practical application requirements of real-time target detection. It will provide new ideas for the intelligent citrus industry.

    Reference
    Related
    Cited by
Get Citation

黄彤镔,黄河清,李震,吕石磊,薛秀云,代秋芳,温威. Citrus fruit recognition method based on the improved model of YOLOv5[J]. Jorunal of Huazhong Agricultural University,2022,41(4):170-177.

Copy
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:January 13,2022
  • Revised:
  • Adopted:
  • Online: October 12,2022
  • Published:
Article QR Code