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

权利要求书

1.一种陶瓷洁具的款式识别方法，其特征在于，包括：

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

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

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

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

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

2.根据权利要求1所述陶瓷洁具的款式识别方法，其特征在于：所述将训练数据输入款式识别模型得到陶瓷洁具的定位位置以及款式类别，包括：

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

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

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

3.根据权利要求2所述陶瓷洁具的款式识别方法，其特征在于：所述对所述默认框进行补偿处理，得到定位位置，包括：

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

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

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

根据所述默认框当前的宽度以及所述默认框的宽度补偿值，确定所述默认框补偿后的宽度；

根据所述默认框当前的高度以及所述默认框的高度补偿值，确定所述默认框补偿后的高度。

4.根据权利要求2所述陶瓷洁具的款式识别方法，其特征在于：所述对每一所述默认框进行预测得到预测结果，包括：

对每一所述默认框进行预测，确定每一所述默认框对应的多个类别概率；所述类别概率表征不同款式的概率；

根据所述类别概率，通过非极大值抑制确定最终保留的默认框，将最终保留的默认框中最大的类别概率对应的款式类别作为最终保留的默认框的预测结果。

5.根据权利要求2所述陶瓷洁具的款式识别方法，其特征在于：所述训练数据包括第一标签和第二标签，所述根据所述定位位置、所述款式类别以及目标损失函数对款式识别模型进行训练，得到训练后的款式识别模型，包括：

根据所述定位位置与所述第一标签的匹配结果，确定匹配值；所述第一标签包括陶瓷洁具的真实位置；

根据所述款式类别以及所述匹配值，采用第一损失函数确定第一损失参数；

根据所述匹配值、所述定位位置、所述第一标签和所述第二标签，采用第二损失函数确定第二损失参数；所述第二标签包括陶瓷洁具的真实款式类别；

将所述第一损失参数、所述第一权重、所述第二损失参数以及所述第二权重进行加权，根据加权结果确定所述目标损失函数的损失值；

根据所述损失值对款式识别模型进行训练，对款式识别模型的模型参数进行更新，得到训练后的款式识别模型。

6.根据权利要求5所述陶瓷洁具的款式识别方法，其特征在于：所述根据所述定位位置与所述第一标签的匹配结果，确定匹配值，包括：

当所述定位位置与所述第一标签的交并比大于等于交并阈值，确定所述匹配值为1；

或者，

当所述定位位置与所述第一标签的交并比小于交并阈值，确定所述匹配值为0。

7.根据权利要求6所述陶瓷洁具的款式识别方法，其特征在于：所述根据加权结果确定所述目标损失函数的损失值，包括：

确定所述匹配值为1的默认框的总数量；

根据所述加权结果与所述总数量的比值，确定所述目标损失函数的损失值。