IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0328892.html
   My bibliography  Save this article

Small target detection of bearing surface defects based on improved YOLOv5s

Author

Listed:
  • Haisong Xu
  • Xiaolin Shi
  • Han Zhang
  • Fan Yang
  • Yun Chen

Abstract

In the context of industrial automation, the accurate detection of small defects on bearing surfaces (dents, bruise, scratch) is crucial for the safe operation of equipment. However, traditional detection methods have problems such as insufficient feature extraction for small targets and sensitivity to background interference. Based on this, an improved YOLOv5s small target detection method ECN-YOLOv5s for bearing surface defects is proposed. First, an efficient channel attention mechanism (ECA) is inserted in the layer before the SPPF in the backbone network, which realizes efficient computation of channel attention and reduces redundant computation while decreasing the model complication. Second, using the Content-Aware ReAssembly of Features (CARAFE) module instead of the original nearest-neighbor up-sampling module achieves light weighting while allowing for better aggregation of contextual information within a larger sensory field, which effectively improves the diversity and effectiveness of the model. Finally, an original loss function is replaced with Normalized Wasserstein Distance (NWD) to decrease the susceptibility to small object positional deviations. Experimental results in our homemade dataset indicate that the mean average precision (mAP) of ECN-YOLOv5s for bearing defect identification is 92.7%. This is an improvement of 2.4% versus the former YOLOv5s model. Compared with some mainstream detection models, the modified model in this study also provides better effects. Thus, the model is able to meet the requirements for the detection of small defects in bearings in industry.

Suggested Citation

  • Haisong Xu & Xiaolin Shi & Han Zhang & Fan Yang & Yun Chen, 2025. "Small target detection of bearing surface defects based on improved YOLOv5s," PLOS ONE, Public Library of Science, vol. 20(7), pages 1-21, July.
  • Handle: RePEc:plo:pone00:0328892
    DOI: 10.1371/journal.pone.0328892
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0328892
    Download Restriction: no

    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0328892&type=printable
    Download Restriction: no

    File URL: https://libkey.io/10.1371/journal.pone.0328892?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    More about this item

    Statistics

    Access and download statistics

    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:plo:pone00:0328892. 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: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    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.