[发明专利]一种针对货架场景大图的重复图片搜索方法及装置

说明书

本发明提供了一种针对货架场景大图的重复图片搜索方法及装置，所述方法包括：计算货架查询图片与数据库中货架数据库图片之间的相似度，并筛选出相似度最高的N张相似货架图片；对每一图片进行SKU位置识别，并根据SKU位置识别结果对每一图片生成SKU序列；基于每一图片的SKU序列，分别计算所述货架查询图片与每一相似货架图片的编辑距离；根据所述编辑距离分别计算每一相似货架图片的最终相似度，将最终相似度大于预设的阈值的图片判定为重复图片。本发明对筛选出的候选相似货架图片进一步基于SKU信息进行排序和相似度计算，最后搜索出与查询图片最相似的重复图片。本发明方案能够有效提高货架场景中重复图片的搜索准确性。

技术领域

本发明涉及图像识别技术领域，尤其是涉及一种针对货架场景大图的重复图片搜索方法及装置。

背景技术

图像搜索技术是计算机视觉中一个重要的研究方向。当给定一个查询图片，图像搜索技术会从已有的底库图片中搜索N张相似的图片，然后根据所搜出来的 N张图片进一步排序，找出一张最相似的图片。

目前，现有的图像搜索技术往往应用在单个建筑标记、某个商品小图的搜索应用中。而针对零售货架场景大图的重复图片的搜索，仅仅应用现有的图像搜索技术很难有较好的搜索效果。其原因在于，有很多不同门店的货架外观是标准化的，即使是两个不同货架拍摄的图片相似度也极高，则现有搜索方法容易将不同的两个货架图片误判为重复图片；此外，当拍照人员从不同角度拍摄同一货架，或者拍摄货架的区域位置略有变化时，照片的差异度会比较大。现有的图像搜索技术在搜索出N张相似图片后，进一步排序时很容易将其他标准化的非重复图片排列前面，而实际是同一货架的不同角度图片则被排在后面。

综上所述，在应用现有搜索技术搜索某一货架时，会返回很多相似的但并不是同个货架的图片出来，很难搜索出拍摄角度差异较大或者拍摄区域变化较大的同一货架图。

发明内容

本发明旨在提供一种针对货架场景大图的重复图片搜索方法及装置，以解决上述技术问题，从而能够提高货架场景中重复图片的搜索准确性。

为了解决上述技术问题，本发明提供了一种针对货架场景大图的重复图片搜索方法，包括：

获取货架查询图片，采用预先构建的货架图片搜索模型分别计算所述货架查询图片与预设数据库中每一货架数据库图片之间的相似度，并从所述货架数据库图片中筛选出相似度最高的N张作为相似货架图片；

采用预先构建的货架SKU识别模型分别对所述货架查询图片和N张所述相似货架图片进行SKU位置识别，并根据SKU位置识别结果按预设的顺序分别对所述货架查询图片和N张所述相似货架图片生成SKU序列；

基于每一图片的SKU序列，采用预设的编辑距离计算公式分别计算所述货架查询图片与N张所述相似货架图片的编辑距离；

根据所述编辑距离分别计算所述货架查询图片与N张所述相似货架图片的最终相似度，并从N张所述相似货架图片中筛选出最终相似度最高的M张作为候选重复图片；

若所述候选重复图片的最终相似度大于预设的阈值，则将该候选重复图片判定为重复图片。

进一步地，所述获取货架查询图片，采用预先构建的货架图片搜索模型分别计算所述货架查询图片与预设数据库中每一货架数据库图片之间的相似度，并从所述货架数据库图片中筛选出相似度最高的N张作为相似货架图片，具体包括：

获取货架查询图片，采用预先构建的货架图片搜索模型分别计算所述货架查询图片和预设数据库中每一货架数据库图片的特征向量值；

根据所述特征向量值计算所述货架查询图片与每一所述货架数据库图片之间的特征距离；

基于所述特征距离对所述货架数据库图片按从小到大进行排序，并从所述货架数据库图片中筛选出排序靠前的N张作为所述相似货架图片。