[发明专利]一种基于图像找寻商城商品的方法、可读存储介质及计算机设备

说明书

本发明公开了一种基于图像找寻商城商品的方法、可存储介质及计算机设备，所述方法包括：将商品库中的商品图片生成图片向量，得到原始图片生成向量；将商品库中的所述商品图片去掉背景；再将去掉背景之后的图片生成图片向量，得到去背景图片生成向量；将所述原始图片生成向量和所述去背景图片生成向量保存到向量数据库中；将用户上传的图片生成图片向量，得到上传图片生成向量；将所述用户上传的图片去掉背景，再将去掉背景之后的上传的图片生成图片向量，得到去背景上传图片生成向量；应用所述上传图片生成向量和所述去背景上传图片生成向量到所述向量数据库中查找符合相似度阈值的最相似的图片，将满足条件的图片及其对应的商品返回业务方。

技术领域

本发明涉及计算机网络技术领域，特别涉及一种基于图像找寻商城商品的方法。

背景技术

运营同事需要把用户上传的商品与公司商品库中已有的商品做关联，但用户上传的商品文字信息描述不清，不确定该商品应该对应到商品库中哪个商品，所以希望用用户上传的商品图片与商品库中商品图片做比较，从而快速找到相似的商品。

现有技术存在如下缺点：

一般将图片使用某个模型抽取特征生成图片向量，然后存到某中数据库中，有新的图片也生成向量到数据库中检索，找出最相似的图片。

因为模型抽取图片特征时图片不同的背景增加了干扰，两个图片相似度就会比较低，导致不确信两个图片是否是同一商品。

发明内容

本发明实施例提供了一种基于图像找寻商城商品的方法。为了对披露的实施例的一些方面有一个基本的理解，下面给出了简单的概括。该概括部分不是泛泛评述，也不是要确定关键/重要组成元素或描绘这些实施例的保护范围。其唯一目的是用简单的形式呈现一些概念，以此作为后面的详细说明的序言。

根据本发明第一方面提供了一种基于图像找寻商城商品的方法，包括：

S1：将商品库中的商品图片生成图片向量，得到原始图片生成向量；

S2：将商品库中的所述商品图片去掉背景；

S3：再将去掉背景之后的图片生成图片向量，得到去背景图片生成向量；

S4：将所述原始图片生成向量和所述去背景图片生成向量保存到向量数据库中；

S5：将用户上传的图片生成图片向量，得到上传图片生成向量；

S6：将所述用户上传的图片去掉背景，再将去掉背景之后的上传的图片生成图片向量，得到去背景上传图片生成向量；

S7：应用所述上传图片生成向量和所述去背景上传图片生成向量到所述向量数据库中查找符合相似度阈值的最相似的图片，将满足条件的图片及其对应的商品返回业务方。

在一些实施例中，所述生成图片向量的方法为：vgg16模型。

在一些实施例中，所述去掉背景的方法为：rembg背景移除工具。

在一些实施例中，所述向量数据库为milvus向量数据库。

在一些实施例中，所述相似度阈值为0-0.5之间的小数。

在一些实施例中，图片的相识度值越接近0越相似，反之越不像。

根据本发明第二方面提供了一种可读存储介质，所述可读存储介质存储有一个或者多个程序，所述一个或者多个程序可被一个或者多个处理器执行，以实现如本申请第一方面所述的基于图像找寻商城商品的方法。

根据本发明第三方面提供了一种计算机设备，包括处理器和存储器，其中，所述存储器，用于存放计算机程序；所述处理器，用于执行存储在所述存储器上的计算机程序时，实现如本申请第一方面所述的基于图像找寻商城商品的方法的步骤。