A lightweight detection algorithm for steel surface defects based on improved YOLOv5

Mingzhao Song; Yang Li; Shihao Zhang; Xiao Ma

doi:10.1117/12.3004023

11 October 2023 A lightweight detection algorithm for steel surface defects based on improved YOLOv5

Mingzhao Song, Yang Li, Shihao Zhang, Xiao Ma

Proceedings Volume 12800, Sixth International Conference on Computer Information Science and Application Technology (CISAT 2023); 128000E (2023) https://doi.org/10.1117/12.3004023
Event: 6th International Conference on Computer Information Science and Application Technology (CISAT 2023), 2023, Hangzhou, China

Abstract

To address the issue of imbalanced complexity and precision in series models of YOLO, this paper proposes a lightweight object detection algorithm based on YOLOv5s as the baseline model, namely YOLOv5-SFD. Firstly, the PANet structure of YOLOv5s is simplified to the basic FPN, and a Locally Dense Connected Convolution (LDCC) is proposed to improve the problem of feature information loss during downsampling by cheap feature extraction and feature reuse. Secondly, the C3 module of the original model is improved based on LDCC and combined with the C3Ghost module to reduce the weight of the model. Finally, a bottleneck residual block, BLS (Bottleneck LDCC-SE), based on LDCC and Squeeze-and-Excitation attention is introduced to enhance the feature extraction capability of the network. On the dataset NEU-DET, compared to the baseline model, the precision, recall, F1 score, mAP@.5 and mAP@.5:.95 is 73.4%, 84.7%, 77%, 91.5% and 52.5%, respectively, which represents an improvement of 3.2%, 2.2%, 2%, 11.8%, and 4.5%. In addition, the number of parameters is reduced by approximately 47%, and the model size is only 55% of the baseline model. The experimental results demonstrate that YOLOv5-SFD has lower model complexity and higher detection accuracy, which meets the requirements of lightweight networks.

(2023) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Mingzhao Song, Yang Li, Shihao Zhang, and Xiao Ma "A lightweight detection algorithm for steel surface defects based on improved YOLOv5", Proc. SPIE 12800, Sixth International Conference on Computer Information Science and Application Technology (CISAT 2023), 128000E (11 October 2023); https://doi.org/10.1117/12.3004023

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
6 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Detection and tracking algorithms

Object detection

Convolution

Data modeling

Defect detection

Algorithm development

Education and training

Show All Keywords

Keywords/Phrases

Search In:

Publication Years