Continuous casting billet defect detection based on improved YOLOv5

Wenhong Liu; Dexin Wang

doi:10.1117/12.3036011

16 August 2024 Continuous casting billet defect detection based on improved YOLOv5

Wenhong Liu, Dexin Wang

Proceedings Volume 13230, Third International Conference on Machine Vision, Automatic Identification, and Detection (MVAID 2024); 132301C (2024) https://doi.org/10.1117/12.3036011
Event: Third International Conference on Machine Vision, Automatic Identification and Detection, 2024, Kunming, China

Abstract

During the production of continuous cast steel, under the constraints of various factors such as material qualities and processing techniques, various types of defects may occur within the steel, such as inter-cracks and shrinkage holes, which will affect the quality of the steel. Continuous casting billet defect detection is different from the general target detection task, which involves small targets and targets with very high aspect ratios. To address the target characteristics, this paper uses improvements based on the YOLOv5 algorithm to improve the accuracy of continuous casting billet defect detection. Firstly, the self-constructed dataset is analysed, while a clustering algorithm based on differential evolutionary algorithm is applied to re-cluster the anchor frames of small targets and extreme aspect ratio targets. For small targets, the attention mechanism and BiFPN structure are used to improve feature extraction. Finally, varifocal loss is introduced to mitigate the effect of positive and negative sample ratio imbalance on the training results. The experimental results show that the mAP of the improved algorithm is improved by 5.1% compared to the original YOLOv5.

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

Citation Download Citation

Wenhong Liu and Dexin Wang "Continuous casting billet defect detection based on improved YOLOv5", Proc. SPIE 13230, Third International Conference on Machine Vision, Automatic Identification, and Detection (MVAID 2024), 132301C (16 August 2024); https://doi.org/10.1117/12.3036011

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
8 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

Defect detection

Evolutionary algorithms

Feature extraction

Feature fusion

Data modeling

Education and training

Show All Keywords

Keywords/Phrases

Search In:

Publication Years