An improved YOLOv5n algorithm for detecting surface defects in industrial components

一种改进的YOLOv5n算法,用于检测工业部件的表面缺陷

阅读:2

Abstract

Due to the small defect areas and indistinct features on industrial components, detecting surface defects with high accuracy remains challenging, often leading to false detections. To address these issues, this paper proposes an improved YOLOv5n algorithm for industrial surface defect detection. The main improvements are as follows: the DSConv-CA module in the backbone network enhances the feature extraction capability, the Gold-YOLO structure replaces the original PANet structure in the neck to improve information fusion, and the SIoU loss function is adopted to replace the regression loss, further improving detection accuracy. Experimental results demonstrate that the improved YOLOv5n algorithm achieves a mean average precision of 75.3% on the NEU-DET dataset, which is 4.3% higher than the original model.

特别声明

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

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

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

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