A lightweight algorithm for steel surface defect detection using improved YOLOv8

一种基于改进型YOLOv8的轻量级钢材表面缺陷检测算法

阅读:1

Abstract

In response to the issues of low precision, a large number of parameters and high model complexity in steel surface defect detection, a lightweight algorithm using improved YOLOv8 is proposed. Firstly, GhostNet is utilized as the backbone network in order to reduce the number of model parameters and computational complexity. Secondly, the MPCA (MultiPath Coordinate Attention) attention mechanism is integrated to enhance feature extraction capabilities. Finally, the SIoU (Simplified IoU ) is used to replace the traditional CIoU loss function, which can make the anchor frame more fast and accurate in the regression process, to improve the stability and the robustness of detection. The experimental results indicate that these enhancements have led to a reduction of 37% in calculation amount for the improved YOLOv8n algorithm, a decrease of 32% in parameter count, and an increase in average detection accuracy ( mAP ) by 1.2%. This model achieves a balance between lightweighting and detection accuracy while providing a viable solution for deployment in computationally resource-constrained edge computing environments such as embedded systems and mobile devices.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。