图像识别模型训练方法、装置、设备及介质
基本信息
申请号 | CN202110701766.2 | 申请日 | - |
公开(公告)号 | CN113505820A | 公开(公告)日 | 2021-10-15 |
申请公布号 | CN113505820A | 申请公布日 | 2021-10-15 |
分类号 | G06K9/62(2006.01)I;G06K9/46(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 宋丽;刘利 | 申请(专利权)人 | 北京阅视智能技术有限责任公司 |
代理机构 | 北京东方亿思知识产权代理有限责任公司 | 代理人 | 赵秀芹 |
地址 | 100085北京市海淀区上地创业中路28号1号楼2层201 | ||
法律状态 | - |
摘要
摘要 | 本申请公开了一种图像识别模型训练方法、装置、设备及介质。其中,图像识别模型训练方法,包括:建立初始图像识别模型;初始图像识别模型中具有损失函数;初始图像识别模型包括特征提取网络与分类网络,其中,特征提取网络用于接收携带标签的样本图像,并输出特征图;分类网络用于接收特征图,并根据特征图中的K个目标特征点输出初始识别结果;在使用训练样本集对初始图像识别模型进行训练的过程中,按照预设衰减规则更新K的值,按照损失函数的损失值更新初始图像识别模型的网络参数;并在损失函数的损失值小于损失阈值的情况下,获得训练完成的图像识别模型。本申请实施例可以有效提高训练完成的图像识别模型的识别能力。 |
