[发明专利]一种SKU图像分类方法、装置、电子设备及存储介质

权利要求书

1.一种SKU图像分类方法，其特征在于，包括：

利用SPU匹配方法聚合同款SPU的SPU图像；其中，任一款所述SPU包括至少两种SKU；

将各个SKU图像中SKU所在区域作为mask；其中，各个所述SKU图像均包含于同一款所述SPU对应的SPU图像；所述将各个SKU图像中SKU所在区域作为mask，包括：

利用图像分割算法提取各个所述SKU图像中SKU相关的第一前景区域；

采用分割效果分算法判断所述第一前景区域是否满足第一预设条件；

若所述第一前景区域满足所述第一预设条件，则将所述第一前景区域作为所述mask；

若所述第一前景区域不满足所述第一预设条件，则判断各个所述SKU图像是否为纯色图；

若任一所述SKU图像为所述纯色图，则根据所述SKU图像的背景色进行二值化处理，并将二值化处理后的第一前景区域的最大连通域作为所述mask；

若任一所述SKU图像不为所述纯色图，则利用关键点检测算法获取各个所述SKU图像中SKU相关的第二前景区域；

采用关键点检测效果分算法判断所述第二前景区域是否满足第二预设条件；

若所述第二前景区域满足所述第二预设条件，则将所述第二前景区域作为所述mask；

若所述第二前景区域不满足所述第二预设条件，则分别利用显著性检测算法和物体检测算法获取各个所述SKU图像中SKU相关的第三前景区域和第四前景区域；

将所述第三前景区域和所述第四前景区域的交集区域作为所述mask；

根据各个所述mask确定对应的颜色信息，并计算各个所述SKU图像之间的颜色距离；

若所述颜色距离小于预设阈值，则确定对应的两个SKU图像为同种SKU图像。

2.根据权利要求1所述的SKU图像分类方法，其特征在于，所述利用SPU匹配方法聚合同款SPU的SPU图像，包括：

利用感知哈希方法或深度学习方法聚合同款SPU的SPU图像。

3.根据权利要求1所述的SKU图像分类方法，其特征在于，所述根据各个所述mask确定对应的颜色信息，并计算各个所述SKU图像之间的颜色距离，包括：

利用颜色聚类方法聚类出所述mask中出现频率最高的三种颜色，并保存对应的颜色出现频率；

将三种所述颜色及对应的颜色出现频率作为所述mask的所述颜色信息；

根据所述颜色信息，计算任意两个所述SKU图像之间的颜色距离。