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

说明书

本申请所提供的一种SKU图像分类方法，包括：利用SPU匹配方法聚合同款SPU的SPU图像；其中，任一款SPU包括至少两种SKU；将各个SKU图像中SKU所在区域作为mask；其中，各个SKU图像均包含于同一款SPU对应的SPU图像；根据各个mask确定对应的颜色信息，并计算各个SKU图像之间的颜色距离；若颜色距离小于预设阈值，则确定对应的两个SKU图像为同种SKU图像。该方法解决了相关技术中无法分类背景不相同的SKU的问题，能够在保证精确度的前提下提高了召回率。此外，该方法利用SPU匹配方法，降低SKU图像分类范围，降低了分类的总体时间开销。本申请还提供一种SKU图像分类装置、电子设备及计算机可读存储介质，均具有上述有益效果。

技术领域

本申请涉及深度学习/计算机视觉技术领域，特别涉及一种SKU图像分类方法、装置、电子设备及计算机可读存储介质。

背景技术

SKU分类的任务是将多个商家提供的相同SKU聚合到一起，常用的技术是图像匹配技术。具体而言是指，根据某个SKU的图像，将其与其他网络商家提供的SKU图像进行匹配，若图像相同，则认为这些商家提供相同的SKU，并将这些SKU聚合在一起；否则，则认为该商家并不提供与其他商家相同的SKU。SKU:全称为Standard Keeping Unit，用于描述一个商品的规格、颜色、款式。比如银色iphone xs就是一个SKU。

目前，SKU分类项目一般采用传统的图像处理技术，例如感知哈希方法或者深度学习方法，判断两个SKU图片是否相同。若相同，则认为是同种SKU；反之亦然。

在现有的SKU分类应用场景下，精确度和召回率需要同时被保证。现有方法例如感知哈希方法或者深度学习方法，直接使用SKU图像做图像级别的匹配，会使大量背景不相同的SKU无法分类到一起。

因此，如何在保证精确度的前提下，召回更多的同种SKU是本领域技术人员亟需解决的技术问题。

发明内容

本申请的目的是提供一种SKU图像分类方法、装置、电子设备及计算机可读存储介质，能够在保证精确度的前提下，召回更多的同种SKU。

为解决上述技术问题，本申请提供一种SKU图像分类方法，包括：

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

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

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

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

优选地，所述利用SPU匹配方法聚合同款SPU的SPU图像，包括：

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

优选地，所述将各个SKU图像中SKU所在区域作为mask，包括：

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

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

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

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

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