Opto-Electronic Engineering, Volume. 51, Issue 5, 240044(2024)

Improvement of GBS-YOLOv7t for steel surface defect detection

Liming Liang... Pengwei Long*, Baohe Lu and Renjie Li |Show fewer author(s)
Author Affiliations
  • School of Electrical Engineering and Automation, Jiangxi University of Science and Technology, Ganzhou, Jiangxi 341000, China
  • show less

    Given that small targets are predominant in the steel surface defect areas, most existing methods cannot balance the trade-off between detection accuracy and speed. In this paper, we propose a steel surface defect detection algorithm based on YOLOv7-tiny (GBS-YOLOv7t). Firstly, we design the GAC-FPN network to fully integrate the target semantic information progressively and across layers, aiming to address the limited information flow issue in traditional feature pyramids. Secondly, we embed a bi-level routing attention (BRA) module to endow the model with dynamic query and sparse perception capabilities, thus enhancing the detection accuracy of small targets. Thirdly, we introduce the SIoU loss function to improve the training and inference capabilities of the model, and to enhance the network robustness. Experimental validation on the public dataset NEU-DET demonstrates an mAP of 72.9% and a precision of 69.9% for GBS-YOLOv7t, achieving improvements of 4.2% and 8.5%, respectively, over the original YOLOv7-tiny model. The FPS reaches 104.1 frames, indicating strong real-time performance. Compared to other detection algorithms, GBS-YOLOv7t is more effective in detecting small targets in steel surface areas, with experimental results showing that the improved algorithm better balances the detection accuracy and speed.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Liming Liang, Pengwei Long, Baohe Lu, Renjie Li. Improvement of GBS-YOLOv7t for steel surface defect detection[J]. Opto-Electronic Engineering, 2024, 51(5): 240044

    Download Citation

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

    Category: Article

    Received: Feb. 29, 2024

    Accepted: Mar. 22, 2024

    Published Online: Jul. 31, 2024

    The Author Email: Long Pengwei (龙鹏威)

    DOI:10.12086/oee.2024.240044

    Topics