【技术实现步骤摘要】
本专利技术涉及图像处理领域,特别涉及一种图片检索方法和装置。
技术介绍
大数据时代,人们可以访问获取的信息资源呈现出爆炸式的增长,互联网上每天都诞生海量的图像信息。为了有效地组织、查询与浏览如此大规模的图像资源,图像检索技术应运而生。卷积神经网络是最近广泛应用于模式识别和计算机视觉等领域的一种算法模型,为增加算法针对真实数据的实际性能测试,进而对应用程序的泛化性提出了更高的要求。特别涉及在泛化性的性能上,神经网络在整个结构设计和计算过程中有大量的时间消耗在样本采集和训练过程上,且在训练过程中需要人工标注,造成人力资源消耗巨大,图片模型的获取过程复杂的结果,增加了实现检索功能的难度。
技术实现思路
有鉴于此,本专利技术实施例提供一种图片检索方法和装置,以解决在图片检索模型的训练过程中需要大量的人工标注,造成人力资源消耗巨大,图片模型的获取过程复杂的结果,检索功能的实现难度增加的问题。第一方面,本专利技术实施例提供了一种图片检索方法,包括:利用图片样本对基模型进行训练得到检索模型;将待检索图片输入到所述检索模型中进行计算得到图片特征,根据所述图片特征进行检索,得到结果图片集。第二方面,本专利技术实施例提供了一种图片检索装置,包括:检索模型获取模块,用于利用图片样本对基模型进行训练得到检索模型;结果图片集获取模块,用于将待检索图片输入到所述检索模型中进行计算得到图片特征,根据所述图片特征进行检索,得到结果图片集。由上述技术方案可知,本专利技术实施例提出一种图片模型训练方法,实现了无需人工标注就可以得到图片检索模型,避免了人力资源的消耗,进而简化了模型训练过程,从而 ...
【技术保护点】
一种图片检索方法,其特征在于,包括:利用图片样本对基模型进行训练得到检索模型;将待检索图片输入到所述检索模型中进行计算得到图片特征,根据所述图片特征进行检索,得到结果图片集。
【技术特征摘要】
1.一种图片检索方法,其特征在于,包括:利用图片样本对基模型进行训练得到检索模型;将待检索图片输入到所述检索模型中进行计算得到图片特征,根据所述图片特征进行检索,得到结果图片集。2.根据权利要求1所述的方法,其特征在于,所述利用图片样本对基模型进行训练得到检索模型,包括:训练或下载基模型,所述基模型用于图像识别;依据场景,修改所述基模型的部分结构得到图片检索网络结构,所述部分结构包括超参数;a、对图片检索网络的非超参数进行初始化,所述初始化包括:与基模型部分相同的网络结构直接使用所述基模型的非超参数,与所述基模型不同的网络结构,使用随机数进行初始化;b、定义图片检索网络结构的损失函数,利用最小化损失函数对神经网络进行优化,从而得到检索预模型;根据不同的所述场景并调整检索预模型结构,重复a-b步骤,得到多个检索预模型并保存到模型库中;将验证集带入至少一个所述检索预模型进行验证得到检索精度,将所述检索精度最高的检索预模型设置为检索模型,所述检索模型用于图像检索。3.根据权利要求2所述的方法,其特征在于,所述定义图片检索网络结构的损失函数,利用最小化损失函数对神经网络进行优化,从而得到检索预模型,包括:c、根据损失的变化,选取一组网络训练参数,所述网络训练参数包括学习率参数和动量参数;d、从样本图片集中随机选取一组样本图片保存到图片库中,并将所述样本图片输入所述图片检索网络结构进行训练操作,所述训练操作包括修改参数,进而最小化损失;重复c、d步骤,直到损失不再明显或持续减小时,则获取检索预模型并保存到模型库中。4.根据权利要求3所述的方法,其特征在于,所述样本图片集包括原始图片组、积极图片组和消极图片组;所述积极图片组是对所述原始图片组中原始图片增强操作后得到的积极图片组成;所述增强操作包括旋转变换、透视变换、反射变换和剪切操作。5.根据权利要求4所述的方法,其特征在于,所述损失函数包括其中,dist(x,y)表示x和y间的距离;表示所述原始图片组中第i张图片输入所述检索预模型得到的样本特征;表示所述积极图片组中第i张图片输入所述检索预模型得到的样本特征;表示所述消极图片组中第i张图片输入所述检索预模型得到的样本特征;α表示与间距离和与间距离的最小间隔;f表示所述检索预模型;f(x)表示图像x通过该检索预模型计算得到的特征。6.根据权利要求1所述的方法,其特征在于,所述将待检索图片输入到所述检索模型中进行计算得到图片特征,根据所述图片特征进行检索,得到结果图片集,包括:将数据库中图片输入所述检索模型,得到图片特征,建立图片和图片特征之间的索引,并将所述图片特征和所述索引存储到特征库;对所述待检索图片进行预处理,并输入到所述检索模型中得到待检索特征;计算所述待检索特征和特征库中的特征间的特征距离,当特征距离小于预设阈值时,从所述图片库中选取与所述特征对应的图片添加到结果图片集,并将所述结果图片集返回。7.根据权利要求5所述的方法,其特征在于,所述特征距离包括欧氏距离、余弦距离或汉明距离。8.一种图片检索装置,其特征在于,包括:检索模型获取模块,用于利用图片样本对基模型进行训练得到检索模型;结果...
【专利技术属性】
技术研发人员:高钰舒,张凯磊,
申请(专利权)人:上海谦问万答吧云计算科技有限公司,
类型:发明
国别省市:上海;31
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。