计算机工程2025，Vol.51Issue(6)：338-348,11.DOI:10.19678/j.issn.1000-3428.0069285
基于TCM-YOLO网络的金属表面缺陷检测方法
Metal Surface Defect Detection Method Based on TCM-YOLO Network
赵小虎 1谢礼逊 1慕灯聪 1张悦1
作者信息
- 1. 中国矿业大学信息与控制工程学院,江苏徐州 221008||矿山互联网应用技术国家地方联合工程实验室(中国矿业大学),江苏徐州 221008
- 折叠
摘要
Abstract
Surface defect detection in metal production and manufacturing suffers from problems of low detection accuracy and slow processing speed.To address these problems,this study proposes a metal defect detection method based on an improved You Only Look Once version 8(YOLOv8)network(TCM-YOLO).This method enhances the coordinate attention mechanism to the Three-Channel Coordinate Attention(TCCA)mechanism and combines it with a second version of the deformable convolutional network,i.e.,the Three-channel Deformable Convolution Network(TDCN),thereby enhancing the feature extraction ability of the network.In the feature fusion network,a bidirectional feature pyramid and Dynamic Snake Convolution(DSC)are combined to improve the missed detection rate in steel strip defect detection,and to improve the retention of tiny texture and complex defect structure information.The Minimum Point Distance Intersection over Union(MPDIoU)loss function is used to replace the original loss function to accelerate the convergence speed and improve regression accuracy.Finally,a global attention mechanism is embedded to continuously capture important information regarding the global shape of the defect.Experimental results show that the average accuracy of the TCM-YOLO algorithm on the steel strip defects dataset of Northeastern University is 81.8％,which is 7.4 percentage points higher than that of the original YOLOv8 algorithm,and the accuracy reaches 78.3％,which is 8.9 percentage points higher than that of the original model.The detection speed of the algorithm reaches 61.73 frame/s.On the Tianchi aluminum profile defect dataset,the average accuracy is 4.1 percentage points higher than that of the original YOLOv8 algorithm and 8.7 percentage points higher than that of the original model.The results show that the TCM-YOLO algorithm has high detection accuracy and fast detection speed,which improves the detection capability for metal surfaces.关键词
缺陷检测/目标检测/YOLOv8算法/注意力机制/损失函数Key words
defect detection/object detection/YOLOv8 algorithm/attention mechanism/loss function分类
信息技术与安全科学引用本文复制引用
赵小虎,谢礼逊,慕灯聪,张悦..基于TCM-YOLO网络的金属表面缺陷检测方法[J].计算机工程,2025,51(6):338-348,11.
