一种快速的多标签图片检索系统及实现方法
基本信息
申请号 | CN201710076634.9 | 申请日 | - |
公开(公告)号 | CN106951911B | 公开(公告)日 | 2021-06-29 |
申请公布号 | CN106951911B | 申请公布日 | 2021-06-29 |
分类号 | G06K9/62(2006.01)I;G06F16/583(2019.01)I | 分类 | 计算;推算;计数; |
发明人 | 胡焜;白洪亮;董远 | 申请(专利权)人 | 北京飞搜科技有限公司 |
代理机构 | 北京知果之信知识产权代理有限公司 | 代理人 | 卜荣丽 |
地址 | 215123 江苏省苏州市苏州工业园区星湖街328号创意产业园2-B702单元 | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种快速的多标签图片检索系统及实现方法,方法包括:在一卷积神经网络中,部署用于提取候选区域的RPN网络,并提取出图片的候选区域信息,对得到的候选区域信息进行ROI池化计算;池化结束后通过一全连接层,再根据多标签信息建立一多标签分类损失函数来训练所述卷积神经网络,同时建立一加权三维损失函数来训练上述的卷积神经网络;通过经过多任务学习后的卷积神经网络在图片候选集中提取每一张图片的哈希码并存入数据库,与数据库中的哈希码对比后,完成图片检索。本发明通过分类与哈希的多任务学习来训练整个网络,从而有效地保证检索的准确性。同时在检索过程中使用汉明距离来度量相似度,大大提升了检索的效率。 |
