基于生成对抗性自动增强网络的图像检索方法及装置.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN113836330A(43)申请公布日2021.12.24(21)申请号202111066975.0(22)申请日2021.09.13(71)申请人清华大学深圳国际研究生院地址518055广东省深圳市南山区西丽街道深圳大学城清华校区A栋二楼(72)发明人方文璇张凯(74)专利代理机构深圳新创友知识产权代理有限公司44223代理人江耀纯(51)Int.Cl.G06F16/53(2019.01)G06F16/583(2019.01)G06F16/55(2019.01)G06N3/04(2006.01)G06N3/08(2006.01)权利要求书2页说明书5页附图2页(54)发明名称基于生成对抗性自动增强网络的图像检索方法及装置(57)摘要本发明公开了基于生成对抗性自动增强网络的图像检索方法，包括对预处理后的街景图像进行划分为训练集街景图像、测试集街景图像和验证集街景图像，将训练集中的街景图像作为输入特征输入生成对抗性自动增强网络，由增强策略网络生成不同的策略，并把这些策略作用于相同的数据上，对数据集中一批大小的图像数据进行策略中的增强处理，得到增强后的数据，然后由目标检索网络，通过学习更多的特征来最小化损失，并更新权重，同时将这些损失收集起来作为反馈更新增强策略网络来最大化损失，得到新的增强策略，并迭代进行，由测试集街景图像和验证集街景图像进行测试和验证得到最终检索结果，以提高图像检索的准确度。CN113836330ACN113836330A权利要求书1/2页1.基于生成对抗性自动增强网络的图像检索方法，其特征在于，包括如下步骤：S1、对公开数据集的街景图像进行预处理；S2、对预处理后的街景图像进行划分为训练集街景图像、测试集街景图像和验证集街景图像；S3、搭建生成包含增强策略网络和目标检索网络的生成对抗性自动增强网络；S4、将训练集中的街景图像作为输入特征输入生成对抗性自动增强网络，由增强策略网络生成不同的策略，并把这些策略作用于相同的数据上，对数据集中一批大小的图像数据进行策略中的增强处理，得到增强后的数据，然后由目标检索网络，通过学习更多的特征来最小化损失，并更新权重，同时将这些损失收集起来作为反馈更新增强策略网络来最大化损失，得到新的增强策略，并迭代进行；S5、将测试集街景图像和验证集街景图像输入至训练与确定好模型参数的生成对抗性自动增强网络，进行测试和验证得到最终检索结果。2.根据权利要求1所述的基于生成对抗性自动增强网络的图像检索方法，其特征在于，步骤S1包括如下步骤：S1‑1、获取公开数据集的街景图像；S1‑2、将公开数据集的街景图像的眼底图像进行裁剪，裁剪成统一的分辨率图像；S1‑3、对统一分辨率的图像进行亮度、对比度、饱和度和色调处理，然后进行归一化操作。3.根据权利要求1所述的基于生成对抗性自动增强网络的图像检索方法，其特征在于，步骤S2中划分为训练集街景图像时，训练集中的每组街景图像包括一个查询图像、一个最简单的正图像、前n个困难的正图像和负图像。4.根据权利要求1所述的基于生成对抗性自动增强网络的图像检索方法，其特征在于，步骤S4中包括将输入生成对抗性自动增强网络的训练集街景图像特征进行若干代模型训练，具体包括如下步骤：S4‑1、将第一代模型的输出作为第二代模型的监督，第一代模型通过与相同算法的方案进行训练；S4‑2、训练收敛后，建立并初始化第二代模型：S4‑3、并使用固定的第一代模型进行相似度标签的估计，用以训练第二代模型，以此循环。5.根据权利要求1所述的基于生成对抗性自动增强网络的图像检索方法，其特征在于，在增强策略网络的搜索空间中，一个增强策略被定义为由m个子策略组成，一个子策略包含两个要依次应用的图像操作，每个操作都是一个图像处理函数，有两个对应的参数，以及应用这个操作的概率和幅度，然后将每个最佳策略连接起来形成一个包含m×m个子策略的单个策略。6.根据权利要求5所述的基于生成对抗性自动增强网络的图像检索方法，其特征在于，对所有操作的幅度都设置在预设的范围值内。7.一种基于生成对抗性自动增强网络的图像检索装置，其特征在于，包括：包括至少一个存储器以及至少一个处理器；所述存储器，包括存储于其中的至少一个可执行程序；2CN113836330A权利要求书2/2页所述可执行程序在由所述处理器执行时，实现如权利要求1至6中任一项所述的方法。3CN113836330A说明书1/5页基于生成对抗性自动增强网络的图像检索方法及装置技术领域[0001]本发明涉及图像检索技术领域，具体为一种基于生成对抗性自动增强网络的图像检索方法及装置。背景技术[0002]现今图像数据海量增长，希望以最小的人力和物力实现快速且精确的图像检索一直是一项重要的