发明公开
- 专利标题: 一种模型训练方法、装置、电子设备及存储介质
-
申请号: CN202311292060.0申请日: 2023-10-08
-
公开(公告)号: CN117475253A公开(公告)日: 2024-01-30
- 发明人: 林容泰 , 黄世华 , 何龙泉 , 沈西
- 申请人: 英特灵达信息技术(深圳)有限公司
- 申请人地址: 广东省深圳市南山区粤海街道软件产业基地4栋B座602室
- 专利权人: 英特灵达信息技术(深圳)有限公司
- 当前专利权人: 英特灵达信息技术(深圳)有限公司
- 当前专利权人地址: 广东省深圳市南山区粤海街道软件产业基地4栋B座602室
- 代理机构: 北京柏杉松知识产权代理事务所
- 代理商 孙翠贤; 孟维娜
- 主分类号: G06V10/774
- IPC分类号: G06V10/774 ; G06V10/764 ; G06V10/22 ; G06V10/40 ; G06V10/26
摘要:
本发明提供一种模型训练方法、装置、电子设备及存储介质,涉及计算机视觉技术领域。该方法包括:确定待训练的图像分类模型,图像分类模型包含特征提取层、多个检测网络与分割网络,每一检测网络对应的目标尺寸范围为该检测网络所检测对象的尺寸范围;基于样本图像与第一标注信息对特征提取层与检测网络训练,得到训练完成的特征提取层与检测网络;特征提取层提取多种尺寸的特征图,检测网络基于属于检测网络尺寸的特征图进行对象检测;基于样本图像、第二标注信息与训练完成的特征提取层对分割网络训练,得到训练完成的分割网络以训练完成图像分类模型。本方案在保证提供不同形式的分类结果的情况下,兼顾模型训练效率的提升以及准确率的提升。