IDEAS home Printed from https://ideas.repec.org/a/gam/jagris/v14y2024i9p1499-d1469371.html
   My bibliography  Save this article

Cotton-YOLO-Seg: An Enhanced YOLOV8 Model for Impurity Rate Detection in Machine-Picked Seed Cotton

Author

Listed:
  • Long Jiang

    (College of Mechanical and Electrical Engineering, Shihezi University, Shihezi 832003, China
    Key Laboratory of Northwest Agricultural Equipment, Ministry of Agriculture, Shihezi 832003, China)

  • Weitao Chen

    (College of Mechanical and Electrical Engineering, Shihezi University, Shihezi 832003, China
    Key Laboratory of Northwest Agricultural Equipment, Ministry of Agriculture, Shihezi 832003, China
    State Key Laboratory of High-Performance Complex Manufacturing, Central South University, Changsha 410083, China)

  • Hongtai Shi

    (College of Mechanical and Electrical Engineering, Shihezi University, Shihezi 832003, China
    Key Laboratory of Northwest Agricultural Equipment, Ministry of Agriculture, Shihezi 832003, China)

  • Hongwen Zhang

    (College of Mechanical and Electrical Engineering, Shihezi University, Shihezi 832003, China
    Key Laboratory of Northwest Agricultural Equipment, Ministry of Agriculture, Shihezi 832003, China)

  • Lei Wang

    (College of Mechanical and Electrical Engineering, Shihezi University, Shihezi 832003, China
    Key Laboratory of Northwest Agricultural Equipment, Ministry of Agriculture, Shihezi 832003, China)

Abstract

The detection of the impurity rate in machine-picked seed cotton is crucial for precision agriculture. This study proposes a novel Cotton-YOLO-Seg cotton-impurity instance segmentation algorithm based on the you only look once version 8 small segmentation model (Yolov8s-Seg). The algorithm achieves precise pixel-level segmentation of cotton and impurities in seed cotton images and establishes a detection model for the impurity rate, enabling accurate detection of the impurity rate in machine-picked cotton. The proposed algorithm removes the Pyramid 4 (P4) feature layer and incorporates Multi-Scale Convolutional Block Attention (MSCBCA) that integrates the Convolutional Block Attention Module (CBAM) and Multi-Scale Convolutional Attention (MSCA) into the Faster Implementation of Cross Stage Partial Bottleneck with 2 Convolutions (C2f) module of the feature extraction network, forming a novel C2f_MSCBCA module. The SlimNeck structure is introduced in the feature fusion network by replacing the P4 feature layer with the small-target detection layer Pyramid 2 (P2). Additionally, transfer learning is employed using the Common Objects in Context (COCO) instance segmentation dataset. The analysis of 100 groups of cotton image samples shows that the Mean Absolute Error (MAE), Root Mean Square Error (RMSE), and Mean Absolute Percentage Error (MAPE) for impurity rate detection are 0.29%, 0.33%, and 3.70%, respectively, which are reduced by 52.46%, 48.44%, and 53.75% compared to the Yolov8s-seg model. The Precision (P), Recall (R), and mean Average Precision at an intersection over union of 0.5 (mAP@0.5) are 85.4%, 78.4%, and 80.8%, respectively, which are improved by 4.2%, 6.2%, and 6.4% compared to Yolov8s-seg model, significantly enhancing the segmentation performance of minor impurities. The Cotton-YOLO-Seg model demonstrates practical significance for precisely detecting the impurity rate in machine-picked seed cotton.

Suggested Citation

  • Long Jiang & Weitao Chen & Hongtai Shi & Hongwen Zhang & Lei Wang, 2024. "Cotton-YOLO-Seg: An Enhanced YOLOV8 Model for Impurity Rate Detection in Machine-Picked Seed Cotton," Agriculture, MDPI, vol. 14(9), pages 1-19, September.
  • Handle: RePEc:gam:jagris:v:14:y:2024:i:9:p:1499-:d:1469371
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2077-0472/14/9/1499/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2077-0472/14/9/1499/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jagris:v:14:y:2024:i:9:p:1499-:d:1469371. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.