基于AlexNet-SVM模型微型钢球表面缺陷快速识别方法
Fast recognition method of surface defects on micro-steel balls based on improved AlexNet-SVM convolutional neural network
栗琳 1王仲 2吴秀丽1
作者信息
- 1. 北京科技大学机械工程学院,北京 100083
- 2. 天津大学精密测试技术及仪器国家重点实验室,天津 300072
- 折叠
摘要
微型钢球由于高反射、球体需要全覆盖的特点,其表面缺陷的质量控制尤为困难.针对人工检测方法效率低且准确度不足的问题,文章提出一种改进的AlexNet的卷积神经网络和SVM模型的钢球表面缺陷快速识别方法.该模型删减了后 3 个卷积层,保留全连接层FC7 提取的特征,采用SVM代替原始Softmax分类器以防止过拟合,提高模型泛化能力.此外,研究了基于K-CV的改进网络搜索算法确定分类器最佳参数.实验采用混淆矩阵对提出模型的识别结果进行性能评估,结果表明,该方法平均准确率达到 99.43%,运算时间为 17.2 ms.对比原模型及其他网络模型,具有较高的准确度和推理速度,能够满足工业现场检测的需求.
Abstract
The quality control of surface defects on micro-steel balls is particularly difficult because of high reflectivity and the need for full coverage on the spherical surface.In response to the problems of low efficiency and inadequate accuracy of manual inspection methods,an improved method is proposed for fast identification of surface defects on steel balls with a combination of an improved AlexNet convolutional neural network and an SVM model.The last three convolutional layers are removed and the features extracted by the fully connected layer FC7 are reserved in our model.The original Softmax classifier is replaced by SVM to prevent overfitting and improve the model's generalization ability.In addition,an improved network search algorithm based on K-CV is used to determine optimal parameters for the classifier.Experimental evaluation of the proposed model's recognition results is performed with a confusion matrix.The results show that this method achieves an average accuracy rate of 99.43%with an operating time of 17.2 ms.Compared to the original model and other network models,it has higher accuracy and inference speed,meeting the requirements of industrial field inspection.
关键词
微型钢球/卷积神经网络/AlexNet/SVM/表面缺陷检测Key words
micro-steel balls/CNN/AlexNet/SVM/surface defect inspection引用本文复制引用
基金项目
国家自然科学基金资助(52175449)
国家重点研发计划(2022YFB3402701)
出版年
2024
NETL
NSTL
万方数据