[发明专利]一种陶瓷洁具的款式识别方法、装置及存储介质

说明书

本发明公开了一种陶瓷洁具的款式识别方法、装置及存储介质，本发明通过获取陶瓷洁具的俯视图、仰视图以及侧视图，然后利用训练后的款式识别模型得到陶瓷洁具的分类结果，结合陶瓷洁具的多个角度的图像，对陶瓷洁具进行分类，不仅能够识别款式单一且外观差异大的陶瓷洁具，也适用于对外观差异不明显或者款式多样的陶瓷洁具的识别，不需要人为识别，提高了效率且降低了成本；根据定位位置、款式类别以及目标损失函数对款式识别模型进行训练，得到训练后的款式识别模型，目标损失函数中款式类别的第一权重大于定位位置的第二权重，使得款式识别模型的训练更关注款式类别，提高训练后的款式识别模型的分类性能，本发明可广泛应用于机器视觉领域。

技术领域

本发明涉及机器视觉领域，尤其是一种陶瓷洁具的款式识别方法、装置及存储介质。

背景技术

陶瓷洁具生产是传统行业，近年来随着制造业自动化、半自动化水平的提高，机器人应用也推广到陶瓷洁具生产行业中。但是由于陶瓷洁具的制作工艺和生产流程特殊性，机器人应用于陶瓷喷釉和打磨等工序上依然存在一定局限性。传统陶瓷洁具生产线具有小批量，混合生产等特点，但目前的机器人喷釉及机器人打磨技术都需要预先对具体一款陶瓷洁具产品示教好机械手运动轨迹和姿态参数，更换产品款式需要载入对应的示教参数。因此，对于一个具体的陶瓷洁具产品，在机器人执行动作前需要先识别该产品的款式型号。

现有的陶瓷产品款式识别方式主要包括条码扫码识别和RFID射频识别。其中，条码扫码识别和RFID射频识别都需要在工序前进行人为识别，再将对应条码或RFID芯片绑定到陶瓷产品上，根本上还是属于人为识别范畴，操作繁琐、效率低。另外，陶瓷品后续涉及烧窑或者喷釉等改变陶瓷产品外观的工序，会破坏条码或RFID芯片，造成后段工序识别失效，且需要企业不断投入耗材成本，成本高。

发明内容

有鉴于此，为了解决上述技术问题，本发明的目的是提供一种陶瓷洁具的款式识别方法、装置及存储介质，提高分类效率以及降低成本。

发明采用的技术方案是：

一种陶瓷洁具的款式识别方法，包括：

获取陶瓷洁具的图像集合；所述图像集合包括所述陶瓷洁具的俯视图、仰视图以及侧视图；

将所述图像集合输入至训练后的款式识别模型，得到分类结果；

训练后的款式识别模型的获取步骤包括：

将训练数据输入款式识别模型得到陶瓷洁具的定位位置以及款式类别；

根据所述定位位置、所述款式类别以及目标损失函数对款式识别模型进行训练，得到训练后的款式识别模型；所述目标损失函数中所述款式类别的第一权重大于所述定位位置的第二权重。

进一步，所述将训练数据输入款式识别模型得到陶瓷洁具的定位位置以及款式类别，包括：

将所述训练数据输入款式识别模型进行特征提取，得到多个不同尺度下的特征图；

在每一所述特征图上生成多个不同尺寸大小的默认框，对每一所述默认框进行预测得到预测结果；所述默认框用于定位陶瓷洁具，所述预测结果包括款式类别；

对所述默认框进行补偿处理，得到定位位置。

进一步，所述对所述默认框进行补偿处理，得到定位位置，包括：

获取所述默认框的补偿误差；所述补偿误差包括所述默认框的中心点横坐标补偿值、所述默认框的中心点纵坐标补偿值、所述默认框的宽度补偿值和所述默认框的高度补偿值；

根据所述默认框当前的中心点横坐标，以及所述默认框当前的宽度与所述默认框的宽度补偿值的乘积，确定所述默认框补偿后的中心点横坐标；