Research on steel profiles surface defect detection algorithm based on machine vision

Shiyi Li; Zemin Fu; Zhifan Zhao

doi:10.1117/12.3038566

23 August 2024 Research on steel profiles surface defect detection algorithm based on machine vision

Shiyi Li, Zemin Fu, Zhifan Zhao

Proceedings Volume 13250, Fourth International Conference on Image Processing and Intelligent Control (IPIC 2024); 132502K (2024) https://doi.org/10.1117/12.3038566
Event: 4th International Conference on Image Processing and Intelligent Control (IPIC 2024), 2024, Kuala Lumpur, Malaysia

Abstract

Steel profiles is widely used in many different fields and is a fundamental raw material in the industrial production process. However, surface defects often occur during the steel production process due to differences in testing environments and production raw materials. Therefore, it is crucial to detect surface defects in steel profiles. However, with the progress of society, science and technology, traditional machine vision inspection methods are increasingly difficult to meet the requirements of modern industrial production in terms of accuracy and real-time performance in the actual inspection process. In this context, this thesis provides an in-depth analysis of deep learning-based defect detection algorithms and improves the state-of-the-art YOLOv5 object detection algorithm according to the characteristics of steel profiles surface defects in order to improve the detection speed and accuracy. First, the steel dataset is regrouped using the K-means algorithm based on IOU measured distances to generate different sets of anchor frames. Second, MixUp was integrated into the Mosaic data extension to reduce overfitting and improve model generalization. Then, the network structure was improved and an attention module was added to further improve the ability to extract network features.

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

Citation Download Citation

Shiyi Li, Zemin Fu, and Zhifan Zhao "Research on steel profiles surface defect detection algorithm based on machine vision", Proc. SPIE 13250, Fourth International Conference on Image Processing and Intelligent Control (IPIC 2024), 132502K (23 August 2024); https://doi.org/10.1117/12.3038566

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