[发明专利]图像检索方法及装置

说明书

本申请公开了一种图像检索方法及装置，属于内容追踪技术领域，图像检索方法包括：获取待检测图像；通过目标检测网络对待检测图像进行检测，以获取待检测图像中嵌入的多个待检测子图像；通过全局提取模型提取各个待检测子图像的全局特征，全局特征可表征待检测子图像经过多种图像转换方式后的各种可能的形式；通过聚类检索在数据库中查找出与待检测子图像的全局特征相匹配的多张相似图像；通过局部特征匹配模型对待检测子图像与相应的多张相似图像进行局部特征匹配；通过对全局特征匹配结果与局部特征匹配结果进行加权处理，重新对多张相似图像进行排序。可以有效地应对用户偷偷溜过检测系统的行为，实现更加快速、精准地图像检索。

技术领域

本申请属于内容追踪领域，具体涉及一种图像检索方法及装置。

背景技术

内容追踪是一项在当今社交媒体平台上广泛使用的技术，可以用于标记错误信息、操纵广告以及执行版权保护等任务。具体地，可以快速响应病毒或者紧急攻击性内容。

现有技术所面临的一个显而易见的挑战是，每天在社交媒体平台上上传的照片数量通常是百万级别甚至数亿级别。想要完全依赖人工审核是一件不可能的事情。因此，必须依靠算法来帮助自动标记或删除不良内容。

此外，由于上传的照片可能是用户想要偷偷溜过检测系统的新尝试，例如添加表情，添加水印，裁剪，嵌入其他照片等等。考虑到图片的量级以及对检测时间的要求，如何设计一套又准又快的图像检索方案是一件非常值得研究的问题。

发明人在研究过程中发现现有技术当前至少存在以下问题：现有内容追踪技术存在难以应对用户添加表情、添加水印、裁剪、嵌入其他照片等偷偷溜过检测系统的行为，难以快速、精准地检索出目标图像。

发明内容

本申请实施例的目的是提供一种图像检索方法和装置，能够解决目前现有内容追踪技术存在难以应对用户添加表情、添加水印、裁剪、嵌入其他照片等偷偷溜过检测系统的行为，难以快速、精准地检索出目标图像的技术问题。

为了解决上述技术问题，本申请是这样实现的：

第一方面，本申请实施例提供了一种图像检索方法，包括：

获取待检测图像；

通过目标检测网络对待检测图像进行检测，以获取待检测图像中嵌入的多个待检测子图像；

通过全局提取模型提取各个待检测子图像的全局特征，全局特征可表征待检测子图像经过多种图像转换方式后的各种可能的形式；

通过聚类检索在数据库中查找出与待检测子图像的全局特征相匹配的多张相似图像；

通过局部特征匹配模型对待检测子图像与相应的多张相似图像进行局部特征匹配；

通过对全局特征匹配结果与局部特征匹配结果进行加权处理，重新对多张相似图像进行排序。

进一步地，图像检索方法还包括：

通过将不同的初始图像嵌入到截图中得到测试图像，或者，将一张初始图像进行不同的掩饰处理后嵌入到截图中得到测试图像；

通过测试图像以及测试图像中的初始图像的坐标训练目标检测网络。

进一步地，图像检索方法还包括：

通过对第一源图像进行多种图像转换方式得到转换后的第一图像集合，通过对第二源图像进行多种图像转换方式得到转换后的第二图像集合，其中，图像转换方式包括：旋转、缩放、裁剪、增加滤镜、添加水印、涂鸦、贴图或者马赛克；

通过转换后的第一图像集合与第二图像集合训练全局提取模型。

进一步地，通过局部特征匹配模型对待检测子图像与相应的多张相似图像进行局部特征匹配，具体包括：