一种基于改进YOLOv3的皮革瑕疵检测方法及系统

摘要

本发明提出一种基于改进YOLOv3的皮革瑕疵检测方法及系统，涉及皮革瑕疵检测的技术领域，首先构建样本皮革瑕疵图像数据集，并对数据集中的皮革瑕疵进行边框标注，将数据集划分为训练集、验证集和测试集；然后构建YOLOv3目标神经网络模型，基于kmeans聚类算法及标签平滑技术改进模型，利用训练集对改进的YOLOv3目标神经网络模型进行训练，并通过验证集验证训练中的YOLOv3目标神经网络模型的性能，得到训练好的YOLOv3目标神经网络模型；最后基于训练好的模型，对测试集图像数据进行皮革瑕疵检测。本发明通过改进YOLOv3提高了模型的泛化性，降低了瑕疵误检漏检概率，进一步提高了皮革瑕疵检测的准确率。