Lightweight forged part surface crack detection algorithm

Home > Archive>Volume 47, Issue 11, 2024 >123-130

Lightweight forged part surface crack detection algorithm
DOI:
                        
                    
CSTR:
                        [cstr]
                    
Author:
                        Zhang ShangZhang Shang
1.Hubei Key Laboratory of Intelligent Vision Based Monitoring for Hydroelectric Engineering, China Three Gorges University， Yichang 443002, China； 2.Hubei Province Engineering Technology Research Center for Construction Quality Testing Equipment, China Three Gorges University，Yichang 443002, China； 3.College of Computer and Information Technology, China Three Gorges University, Yichang 443002,China
Find this author on All Journals
Find this author on BaiDu
Search for this author on this site
Xu HuanXu Huan
1.Hubei Key Laboratory of Intelligent Vision Based Monitoring for Hydroelectric Engineering, China Three Gorges University， Yichang 443002, China； 2.Hubei Province Engineering Technology Research Center for Construction Quality Testing Equipment, China Three Gorges University，Yichang 443002, China； 3.College of Computer and Information Technology, China Three Gorges University, Yichang 443002,China
Find this author on All Journals
Find this author on BaiDu
Search for this author on this site
Zhang YueZhang Yue
1.Hubei Key Laboratory of Intelligent Vision Based Monitoring for Hydroelectric Engineering, China Three Gorges University， Yichang 443002, China； 2.Hubei Province Engineering Technology Research Center for Construction Quality Testing Equipment, China Three Gorges University，Yichang 443002, China； 3.College of Computer and Information Technology, China Three Gorges University, Yichang 443002,China
Find this author on All Journals
Find this author on BaiDu
Search for this author on this site

                    
Affiliation:1.Hubei Key Laboratory of Intelligent Vision Based Monitoring for Hydroelectric Engineering, China Three Gorges University， Yichang 443002, China; 2.Hubei Province Engineering Technology Research Center for Construction Quality Testing Equipment, China Three Gorges University，Yichang 443002, China; 3.College of Computer and Information Technology, China Three Gorges University, Yichang 443002,China
Clc Number:TP391.4; TN98
Fund Project:

Article

Figures

Metrics

Reference

Cited by

Materials

Comments

Abstract:

To address issues of high memory usage, computational complexity, and inadequate detection speed in defect detection algorithms for complex scenarios, this paper proposes a lightweight forged defect detection algorithm based on YOLOv8. First, magnetic particle inspection images from the production line of heavy truck steering knuckles were collected to construct a forged surface crack dataset. Then, a lightweight convolution module, GSConvns, was introduced to enhance feature interaction and reduce computational load. The Shape-IOU loss function was employed to optimize training performance. Finally, the LAMP pruning strategy was used to remove unnecessary weight parameters, reducing model size and increasing detection speed. Experimental results show that the model achieves a mAP of 83.8%, with parameter and computational reductions of 85.05% and 80.25%, respectively. Detection speed improved from 38.7 FPS to 65.6 FPS, significantly outperforming other mainstream algorithms, making it more suitable for real-time detection. The algorithm′s generalization capability was further verified on a public dataset, with the unpruned improved algorithm′s mAP value increasing by 2.0% compared to the baseline. In summary, this algorithm significantly enhances detection speed and resource efficiency without substantially compromising detection accuracy.

Key words:surface defect detection;YOLOv8 algorithm;lightweight model;loss function;model pruning

Get Citation

Copy

Article Metrics

Abstract:
PDF:
HTML:
Cited by:

History

Received:
Revised:
Adopted:
Online: October 12,2024
Published:

Home

Introduction

Editorial Committee

Policy

Contact Us

中文版

Get Citation

Share

Article Metrics

History

Article QR Code