IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v12y2024i7p957-d1362645.html
   My bibliography  Save this article

SAB-YOLOv5: An Improved YOLOv5 Model for Permanent Magnetic Ferrite Magnet Rotor Detection

Author

Listed:
  • Bo Yu

    (Artificial Intelligence Research Institute, China University of Mining and Technology, Xuzhou 221116, China
    College of Information and Control Engineering, China University of Mining and Technology, Xuzhou 221116, China)

  • Qi Li

    (School of Computer and Communication Engineering, Naning Tech University Pujiang Institute, Naning 211112, China)

  • Wenhua Jiao

    (Artificial Intelligence Research Institute, China University of Mining and Technology, Xuzhou 221116, China
    College of Information and Control Engineering, China University of Mining and Technology, Xuzhou 221116, China)

  • Shiyang Zhang

    (Artificial Intelligence Research Institute, China University of Mining and Technology, Xuzhou 221116, China
    College of Information and Control Engineering, China University of Mining and Technology, Xuzhou 221116, China)

  • Yongjun Zhu

    (Artificial Intelligence Research Institute, China University of Mining and Technology, Xuzhou 221116, China
    College of Information and Control Engineering, China University of Mining and Technology, Xuzhou 221116, China)

Abstract

Surface defects on the permanent magnetic ferrite magnet rotor are the primary cause for the decline in performance and safety hazards in permanent magnet motors. Machine-vision methods offer the possibility to identify defects automatically. In response to the challenges in the permanent magnetic ferrite magnet rotor, this study proposes an improved You Only Look Once (YOLO) algorithm named SAB-YOLOv5. Utilizing a line-scan camera, images capturing the complete surface of a general object are obtained, and a dataset containing surface defects is constructed. Simultaneously, an improved YOLOv5-based surface defect algorithm is introduced. Firstly, the algorithm enhances the capability to extract features at different scales by incorporating the Atrous Spatial Pyramid Pooling (ASPP) structure. Then, the fusion of features is improved by combining the tensor concatenation operation of the feature-melting network with the Bidirectional Feature Pyramid Network (BiFPN) structure. Finally, the introduction of the spatial pyramid dilated (SPD) convolutional structure into the backbone network and output end enhances the detection performance for minute defects on the target surface. In the study, the SAB-YOlOv5 algorithm shows an obvious increase from 84.2% to 98.3% in the mean average precision (mAP) compared to that of the original YOLOv5 algorithm. The results demonstrate that the data acquisition method and detection algorithm designed in this paper effectively enhance the efficiency of defect detection permanent magnetic ferrite magnet rotors.

Suggested Citation

  • Bo Yu & Qi Li & Wenhua Jiao & Shiyang Zhang & Yongjun Zhu, 2024. "SAB-YOLOv5: An Improved YOLOv5 Model for Permanent Magnetic Ferrite Magnet Rotor Detection," Mathematics, MDPI, vol. 12(7), pages 1-17, March.
    • Handle: RePEc:gam:jmathe:v:12:y:2024:i:7:p:957-:d:1362645
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/12/7/957/pdf
    Download Restriction: no
    File URL: https://www.mdpi.com/2227-7390/12/7/957/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Gui Yu & Xinglin Zhou, 2023. "An Improved YOLOv5 Crack Detection Method Combined with a Bottleneck Transformer," Mathematics, MDPI, vol. 11(10), pages 1-12, May.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Jae Hyun Yoon & Jong Won Jung & Seok Bong Yoo, 2024. "Auxcoformer: Auxiliary and Contrastive Transformer for Robust Crack Detection in Adverse Weather Conditions," Mathematics, MDPI, vol. 12(5), pages 1-20, February.
    2. Feng Xiao & Haibin Wang & Yueqin Xu & Zhen Shi, 2023. "A Lightweight Detection Method for Blueberry Fruit Maturity Based on an Improved YOLOv5 Algorithm," Agriculture, MDPI, vol. 14(1), pages 1-18, December.

    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:jmathe:v:12:y:2024:i:7:p:957-:d:1362645. 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.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with 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.