← 返回列表

基于对比学习的二值码图像检索方法及系统

申请号: CN202410056492.X
申请人: 山东建筑大学; 山东积成智通新能源有限公司; 积成能源有限公司
申请日期: 2024/1/16

摘要文本

本发明涉及图像检索技术领域,特别是涉及基于对比学习的二值码图像检索方法及系统,其中方法包括:构建训练集,所述训练集包括若干个正样本图像对和若干个负样本图像对;将训练集,输入到图像分类模型中,对模型进行训练,当模型的总损失函数值不再下降时,停止训练,得到训练后的图像分类模型;获取待检索图像和图像数据库,对待检索图像进行预处理,将待检索图像和图像数据库的图像,均输入到训练后的图像分类模型中,从图像数据库中找到与待检索图像最相似的图像。通过对比学习方法不仅学习全局的语义信息,还可以捕捉细粒度的语义信息,对每个图像之间构建成对相似度矩阵,提升模型挖掘数据点之间相似性的能力。

专利详细信息

项目 内容
专利名称 基于对比学习的二值码图像检索方法及系统
专利类型 发明申请
申请号 CN202410056492.X
申请日 2024/1/16
公告号 CN117573915A
公开日 2024/2/20
IPC主分类号 G06F16/583
权利人 山东建筑大学; 山东积成智通新能源有限公司; 积成能源有限公司
发明人 聂秀山; 孟子玉; 刘新锋; 张立伟; 王林先; 孙兴伟
地址 山东省济南市历城区临港开发区凤鸣路1000号; 山东省济南市高新区科航路1677号A栋四楼; 山东省济南市高新区科航路1677号

专利主权项内容

1.基于对比学习的二值码图像检索方法,其特征是,包括:构建训练集,所述训练集包括若干个正样本图像对和若干个负样本图像对;将训练集,输入到图像分类模型中,对模型进行训练,当模型的总损失函数值不再下降时,停止训练,得到训练后的图像分类模型;其中,总损失函数,是均方误差损失函数和泰勒损失函数的求和结果;泰勒损失函数,包括:权重学习损失函数;所述均方误差损失函数,用于保证哈希码与特征之间的平衡;所述泰勒损失函数,用于通过对比学习方式学习图像的全局特征和局部特征;基于图像的全局特征和局部特征,在每个图像对之间构建成对相似性矩阵,并对成对相似性矩阵进行更新;获取待检索图像和图像数据库,对待检索图像进行预处理,将待检索图像和图像数据库的图像,均输入到训练后的图像分类模型中,从图像数据库中找到与待检索图像最相似的图像。