|
|
Defect detection algorithm of strip surface based on STCS-YOLO |
ZHOU Yaluo1, WU Xianchao1, LIU Wenguang2, ZHANG Ruicheng1 |
1. Collage of Electrical Engineering, North China University of Science and Technology, Tangshan 063210, Hebei, China; 2. Shougang Jingtang Limited Iron and Steel Co., Ltd., Tangshan 063200, Hebei, China |
|
|
Abstract Strip steel surface defect detection technology is one of the important technologies for the production of high quality strip steel products. Aiming at the problems of misdetection, inaccurate positioning, and poor detection ability of small-scale defect targets in previous strip surface defect detection, a strip surface defect detection algorithm (STCS-YOLO) of improved YOLOv5 was proposed. Firstly, the Swin Transformer module was used to fuse with the original C3 module in the output part of feature fusion network to enhance the interaction and reuse of global feature information, which significantly improved the detection ability of small-scale defect targets. Secondly, a lightweight upsampling operator CARAFE was used to replace the traditional sampling operation to better recover defect information, which improved the recognition accuracy of strip surface defects. Finally, embedding 3-D weight attention mechanism SimAM in feature extraction network to enhance the ability of focusing on the foreground feature information, which improved the strong identification ability of defect target. The experimental results demonstrate that the proposed algorithm achieves 79.7% of PmA(mean average precision) on the NEU-DET dataset, which is improvement of 3.9 percentage points over the original network. Additionally, while maintaining model weight and computational complexity nearly unchanged, the single-frame detection time reaches 10.9 ms, which can basically meet the requirements of accurate and rapid detection of strip surface defects. The strip surface defect detection algorithm proposed in this paper lays the technological foundation for improving neat and flawless high quality strip products.
|
Received: 24 July 2023
|
|
|
|
[1] |
陈程,安成钢,李文远. 关于热轧宽带钢行业高质量发展的思考和建议[J]. 中国冶金,2020,30(11):1.
|
[2] |
王文广, 徐芳, 李兴波,等. 热连轧高速钢轧辊对产品表面质量和板形控制的影响[J]. 中国冶金, 2022, 32(2):97.
|
[3] |
罗东亮,蔡雨萱,杨子豪,等. 工业缺陷检测深度学习方法综述[J]. 中国科学(信息科学),2022,52(6):1002.
|
[4] |
李江昀,杨志方,郑俊锋,等. 深度学习技术在钢铁工业中的应用[J]. 钢铁,2021,56(9):43.
|
[5] |
王勃,胡冲,王杰,等. 一种新型涡流检测法及其检测效果[J]. 中国冶金,2021,31(2):50.
|
[6] |
杨理践,郑文学,高松巍,等. 基于平衡电磁技术的钢板裂纹缺陷检测方法[J]. 仪器仪表学报,2020,41(10):196.
|
[7] |
QU Z,JIANG P,ZHANG W. Development and application of infrared thermography non-destructive testing techniques[J]. Sensors,2020,20(14): 3851.
|
[8] |
赵朗月,吴一全. 基于机器视觉的表面缺陷检测方法研究进展[J]. 仪器仪表学报,2022,43(1):198.
|
[9] |
REN Z,FANG F,YAN N,et al. State of the art in defect detection based on machine vision[J]. International Journal of Precision Engineering and Manufacturing-Green Technology,2022,9(2): 661.
|
[10] |
陈方升,何纯玉,陈亚飞. 改进Canny算法在中厚板边缘检测中的应用[J]. 轧钢,2021,38(5):81.
|
[11] |
王军,肖晶晶,胡刚雨. 基于改进的2D-OTSU磁柱表面缺陷检测方法[J]. 制造业自动化,2022,44(8):90.
|
[12] |
FANG X, LUO Q, ZHOU B,et al. Research progress of automated visual surface defect detection for industrial metal planar materials[J]. Sensors,2020,20(18): 5136.
|
[13] |
田媛,郭慧,张伟,等. 基于改进YOLOv4算法的结构用锯材表面缺陷检测研究[J]. 制造业自动化,2023,45(4):26.
|
[14] |
沈春光,李虎威,荆涛,等. 基于深度学习的带钢表面缺陷检测在小样本数据集的应用[J]. 轧钢,2022,39(2):82.
|
[15] |
胡欣,周运强,肖剑,等. 基于改进YOLOv5的螺纹钢表面缺陷检测[J]. 图学学报,2023,44(3):427.
|
[16] |
吴昉,王伟,刘卫朋. 结合注意力机制和卷积神经网络的X射线焊缝缺陷检测[J]. 科学技术与工程,2023,23(8):3387.
|
[17] |
邓能辉,侯睿,叶俊明. 基于深度学习的圆钢表面缺陷检测系统[J]. 中国冶金,2022,32(12):113.
|
[18] |
ZENG N,WU P,WANG Z,et al. A small-sized object detection oriented multi-scale feature fusion approach with application to defect detection[J]. IEEE Transactions on Instrumentation and Measurement, 2022,71: 1.
|
[19] |
寇旭鹏,刘帅君,麻之润. 基于Faster-RCNN的钢带缺陷检测方法[J]. 中国冶金,2021,31(4):77.
|
[20] |
戴张杰,黄成永,刘威,等. 基于Mask-RCNN的转炉炉口形貌检测[J]. 钢铁,2023,58(3):73.
|
[21] |
GIRSHICK R. Fast R-CNN[C]//Proceedings of the IEEE international conference on computer vision. Santiago:IEEE,2015: 1440.
|
[22] |
REN S, HE K, GIRSHICK R, et al. Faster R-CNN: Towards real-time object detection with region proposal networks[J]. Advances in neural information processing systems, 2015, 28.
|
[23] |
HE K, GKIOXARI G, DOLLÁR P, et al. Mask R-CNN[C]//Proceedings of the IEEE international conference on computer vision. Venice: IEEE, 2017: 2961.
|
[24] |
LIU W, ANGUELOV D, ERHAN D, et al. Ssd: Single shot multibox detector[C]//Computer Vision-ECCV 2016: 14th European Conference. Amsterdam, Netherlands:Springer International Publishing, 2016: 21.
|
[25] |
REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: Unified, real-time object detection[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. Las Vegas:IEEE,2016: 779.
|
[26] |
REDMON J, FARHADI A. YOLO9000: better, faster, stronger[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. Honolulu:IEEE,2017:7263.
|
[27] |
REDMON J,FARHADI A. YOLOv3: An incremental improvement[J]. arXiv preprint arXiv:1804. 02767, 2018.
|
[28] |
刘群坡,方源,张建军,等. 基于特征融合SSD的微电连接器缺陷检测[J]. 华中科技大学学报(自然科学版),2022,50(3):49.
|
[29] |
程婧怡,段先华,朱伟. 改进YOLOv3的金属表面缺陷检测研究[J]. 计算机工程与应用,2021,57(19):252.
|
[30] |
杜少聪,张红钢,王小敏. 基于改进YOLOv5的钢轨表面缺陷检测[J]. 北京交通大学学报,2023,47(2):129.
|
[31] |
LIN T Y, DOLLÁR P, GIRSHICK R, et al. Feature pyramid networks for object detection[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. Honolulu:IEEE,2017: 2117.
|
[32] |
LIU S, QI L, QIN H, et al. Path aggregation network for instance segmentation[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. Salt Lake City: IEEE,2018: 8759.
|
[33] |
LIU Z, LIN Y, CAO Y, et al. Swin transformer: Hierarchical vision transformer using shifted windows[C]//Proceedings of the IEEE/CVF international conference on computer vision.Montreal:IEEE,2021: 10012.
|
[34] |
WANG J, CHEN K, XU R, et al. Carafe: Content-aware reassembly of features[C]//Proceedings of the IEEE/CVF international conference on computer vision. Seoul:IEEE,2019: 3007.
|
[35] |
YANG L, ZHANG R Y, LI L, et al. Simam: A simple, parameter-free attention module for convolutional neural networks[C]//International conference on machine learning. Virtual:PMLR, 2021: 11863.
|
|
|
|