摘要
针对热轧带钢表面缺陷检测中存在的检测速度慢、检测精度低等问题,提出了一种改进的YOLOv3算法模型.使用加权K-means聚类算法来优化确定先验框参数,提高先验框(priors anchor)与特征图层(feature map)的匹配度;同时,调整YOLOv3算法的网络结构,融合浅层特征与深层特征,形成新的大尺度检测图层,提高网络对带钢表面缺陷的检测精度.实验结果表明,改进后的YOLOv3算法在NEU-DET数据集上平均精度均值达到了80%,较原有的YOLOv3算法提高了11%;同时检测速度保持在50fps,优于目前其它深度学习带钢表面缺陷检测算法.
To solve the problem of slow speed and low accuracy in the surface defect detection of hot rolled strips,an improved YOLOv3 algorithm is proposed.Firstly,the weighting K-means clustering algorithm is put forward to optimize priors anchor’s parameters,which can improve the match between priors anchor and feature map.Secondly,the improved network structure of the YOLOv3 algorithm is proposed to improve the detection accuracy,whose shallow features and deep features are combined to form the new large-scale inspection layer.The experiments are carried out on the NEU-DET dataset,the results show that the average accuracy of the improved YOLOv3 algorithm is 80%,which is 11%higher than that of the original algorithm;the detection speed is 50fps,which is faster than other strip surface defect detection algorithms based on deep learning.
作者
李维刚
叶欣
赵云涛
王文波
LI Wei-gang;YE Xin;ZHAO Yun-tao;WANG Wen-bo(Engineering Research Center for Metallurgical Automation and Technology of Ministry of Education,Wuhan University of Science and Technology,Wuhan,Hubei 430081,China;National-Provincial Joint Engineering Research Center of High Temperature Materials and Lining Technology,Wuhan University of Science and Technology,Wuhan,Hubei 430081,China;College of Science,Wuhan University of Science and Technology Wuhan,Hubei 430081,China)
出处
《电子学报》
EI
CAS
CSCD
北大核心
2020年第7期1284-1292,共9页
Acta Electronica Sinica
基金
国家自然科学基金(No.51774219)。
作者简介
李维刚,男,1977年出生于湖北通城,博士,武汉科技大学信息科学与工程学院教授、博导,主要研究方向:冶金过程控制、人工智能和深度学习新方法,E-mail:liweigang.luck@foxmail.com;叶欣,男,1997年出生于湖北宜昌,武汉科技大学信息科学与工程学院硕士研究生,主要研究方向:工业过程建模与仿真、深度学习;赵云涛,男,1982年出生于内蒙古赤峰,博士,武汉科技大学信息科学与工程学院副教授,主要研究方向:计算机仿真;王文波,男,1978年出生于湖北襄阳,博士,武汉科技大学信息与计算科学系教授,主要研究方向为多尺度分析与电力系统信号处理、深度学习.
