Opto-Electronic Engineering, Volume. 52, Issue 5, 250032(2025)

Steel surface defect detection based on YOLOv8-SOE

Yujie Ma1,2, Chuqing Cao1,2、*, and Jing Zhang2
Author Affiliations
  • 1School of Computer and Information, Anhui Polytechnic University, Wuhu, Anhui 241000, China
  • 2Yangtze River Delta HIT Robot Technology Research Institute, Wuhu, Anhui 241000, China
  • show less

    In order to improve the detection capability of small target defects in steel surface inspection, an improved YOLOv8-SOE model is proposed. The model processes the P2 layer features by designing the FSCConv module. By compressing the P2 layer features and deeply fusing them with the P3 layer features, the model's sensitivity to small target features is effectively enhanced, while avoiding the computational burden caused by the introduction of additional detection layers. In order to further optimize the multi-scale feature fusion capability, cross stage partial omni-kernel (CSP-OK) module is used to optimize the multi-scale feature fusion, which improves the integration efficiency of features of different scales. The SIoU loss function is introduced to optimize the bounding box regression, which further improves the positioning accuracy. Experimental results show that the mAP of the YOLOv8-SOE model on the NEU-DET dataset achieves 80.7%, which is 5.4% higher than the baseline model, and has good generalization ability on the VOC2012 dataset. While improving the accuracy of small target detection, the model maintains a high computational efficiency and has good application prospects.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Yujie Ma, Chuqing Cao, Jing Zhang. Steel surface defect detection based on YOLOv8-SOE[J]. Opto-Electronic Engineering, 2025, 52(5): 250032

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Article

    Received: Feb. 13, 2025

    Accepted: Apr. 3, 2025

    Published Online: Jul. 18, 2025

    The Author Email: Chuqing Cao (曹雏清)

    DOI:10.12086/oee.2025.250032

    Topics