[发明专利]一种识别SKU的训练方法及装置

权利要求书

1.一种识别SKU的训练方法，其特征在于，包括：

获取预设数量的包含有SKU标注信息的图像样本，为所述SKU标注信息构建统一的标注格式；其中，所述标注信息包括框选的SKU选区及SKU信息；所述图像样本包括线下商场拍摄的A类照片以及通过布景摆拍、3D建模和/或网络爬取获得的C类照片；

将标注格式统一后的图像样本按照预设比例随机分配为检测器的训练集和测试集，使得检测器能够对所述图像样本中的SKU进行自动框选得到检测结果；其中，所述检测结果为所述图像样本被框选的区域；

通过对所述检测结果进行特征提取来训练分类器，使得分类器能够根据检测结果自动输出最大概率类别的SKU信息；其中，所述分类器包括一级分类器和二级分类器；所述一级分类器用于根据预设的分类需求对所述检测结果进行粗分类，得到第一分类结果；所述二级分类器用于根据所述第一分类结果对所述检测结果进行细分类。

2.根据权利要求1所述的识别SKU的训练方法，其特征在于，将标注格式统一后的图像样本的70%~90%作为训练集，10%~30%作为测试集对检测器进行训练。

3.根据权利要求1所述的识别SKU的训练方法，其特征在于，所述图像样本的采集场景包括货柜、冰柜、陈列挂架、冷风柜、专柜平台和堆头。

4.根据权利要求1所述的识别SKU的训练方法，其特征在于，所述SKU信息包括商品类别和型号信息。

5.根据权利要求1所述的识别SKU的训练方法，其特征在于，所述分类器包括others类别，用于加入不同的负样本，降低误识别率。

6.一种识别SKU的训练装置，其特征在于，包括：

样本获取模块，用于获取预设数量的包含有SKU标注信息的图像样本，为所述SKU标注信息构建统一的标注格式；其中，所述标注信息包括框选的SKU选区及SKU信息；所述图像样本包括线下商场拍摄的A类照片以及通过布景摆拍、3D建模和/或网络爬取获得的C类照片；

检测器训练模块，用于将标注格式统一后的图像样本按照预设比例随机分配为检测器的训练集和测试集，使得检测器能够对所述图像样本中的SKU进行自动框选得到检测结果；其中，所述检测结果为所述图像样本被框选的区域；

分类器训练模块，用于通过对所述检测结果进行特征提取来训练分类器，使得分类器能够根据检测结果自动输出最大概率类别的SKU信息；其中，所述分类器包括一级分类器和二级分类器；所述一级分类器用于根据预设的分类需求对所述检测结果进行粗分类，得到第一分类结果；所述二级分类器用于根据所述第一分类结果对所述检测结果进行细分类。

7.一种计算机终端设备，其特征在于，包括：

一个或多个处理器；

存储器，与所述处理器耦接，用于存储一个或多个程序；

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如权利要求1至5任一项所述的识别SKU的训练方法。

8.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至5任一项所述的识别SKU的训练方法。