一种基于深度学习的刺绣针法识别方法

本发明提供了一种基于深度学习的刺绣针法识别方法，包括：步骤1，确定针法并收集相关图片，根据需要的针法找到对应的针法图片；步骤2，对图片进行预处理和标注，得到刺绣数据集；步骤3，将刺绣数据集划分为训练集，验证集和测试集，进行配置撰写；步骤4，对YOLO模型进行改进；步骤5，使用训练集训练改进的YOLO模型；步骤6，对改进的YOLO模型进行评估，如果不满足条件，调整模型结构和模型训练参数，返回步骤5；如果满足条件，将测试集输入改进的YOLO模型，得到预测结果。本发明能够在复杂纹理、光照变化和绣制角度多样性的情况下保持较高的识别准确率和鲁棒性，适用于多种刺绣针法。