图像识别模型训练方法、装置、设备及介质

基本信息

申请号 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的值,按照损失函数的损失值更新初始图像识别模型的网络参数;并在损失函数的损失值小于损失阈值的情况下,获得训练完成的图像识别模型。本申请实施例可以有效提高训练完成的图像识别模型的识别能力。