基于AlexNet-SVM模型微型钢球表面缺陷快速识别方法

Fast recognition method of surface defects on micro-steel balls based on improved AlexNet-SVM convolutional neural network

微型钢球由于高反射、球体需要全覆盖的特点,其表面缺陷的质量控制尤为困难。针对人工检测方法效率低且准确度不足的问题,文章提出一种改进的AlexNet的卷积神经网络和SVM模型的钢球表面缺陷快速识别方法。该模型删减了后3个卷积层,保留全连接层FC7提取的特征,采用SVM代替原始Softmax分类器以防止过拟合,提高模型泛化能力。此外,研究了基于K-CV的改进网络搜索算法确定分类器最佳参数。实验采用混淆矩阵对提出模型的识别结果进行性能评估,结果表明,该方法平均准确率达到99.43%,运算时间为17.2 ms。对比原模型及其他网络模型,具有较高的准确度和推理速度,能够满足工业现场检测的需求。

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.