神经网络模型的训练方法及图像检索方法
基本信息
申请号 | CN202111077276.6 | 申请日 | - |
公开(公告)号 | CN113807516A | 公开(公告)日 | 2021-12-17 |
申请公布号 | CN113807516A | 申请公布日 | 2021-12-17 |
分类号 | G06N3/08(2006.01)I;G06K9/62(2006.01)I;G06K9/46(2006.01)I;G06K9/32(2006.01)I;G06F16/583(2019.01)I | 分类 | 计算;推算;计数; |
发明人 | 李东明;金忠良;林赞磊 | 申请(专利权)人 | 长城计算机软件与系统有限公司 |
代理机构 | 北京轻创知识产权代理有限公司 | 代理人 | 朱晓彤 |
地址 | 100190北京市海淀区中关村东路66号甲1号楼17-19层 | ||
法律状态 | - |
摘要
摘要 | 本发明公开了神经网络模型的训练方法及图像检索方法,涉及图像处理技术领域。训练方法包括:基于semi‑hard负例策略选择query图像的正例图像和负例图像,构成三元组;将三元组的图像输入到神经网络模型中,对图像进行不同深度卷积层的特征提取,结合注意力网络从得到的特征图中提取感兴趣区域的局部特征;对得到的局部特征进行聚合,得到全局特征;根据全局特征对比损失函数,并通过反向传播算法更新神经网络模型的权重,直到达到预设的训练停止条件,完成训练。本发明可以有效的降低训练的复杂度,加快训练速度。 |
