- 专利标题: 一种船舶类别分类模型训练方法、系统及计算机存储介质
-
申请号: CN202210525452.6申请日: 2022-05-16
-
公开(公告)号: CN114627338B公开(公告)日: 2022-09-09
- 发明人: 吴显德 , 邹凡 , 金振中 , 王震艳 , 钱茂俊
- 申请人: 浙江华是科技股份有限公司
- 申请人地址: 浙江省杭州市余杭区闲林街道嘉企路16号3幢1楼
- 专利权人: 浙江华是科技股份有限公司
- 当前专利权人: 浙江华是科技股份有限公司
- 当前专利权人地址: 浙江省杭州市余杭区闲林街道嘉企路16号3幢1楼
- 代理机构: 北京城烽知识产权代理事务所
- 代理商 王新月
- 主分类号: G06V10/764
- IPC分类号: G06V10/764 ; G06V10/774 ; G06K9/62
摘要:
本发明公开一种船舶类别分类模型训练方法、系统及计算机存储介质。该方法包括:S1将原始训练集分成多个小数据集;S2.将多个小数据集分为多组训练样本;S3.将每组训练样本进行训练得到一一对应的多个船舶类别分类模型以及多个难例数据集;S4.将多个船舶类别分类模型进行融合得到预训练模型;S5.加载预训练模型并通过数据增强方法对原始训练集和多个难例数据集进行m轮模型训练,得到当前更新的船舶类别分类模型;S6.对原始测试集训练,得到当前更新的难例数据集,对其和原始训练集进行m+1轮模型训练,得到下一更新的船舶类别分类模型,重复S6进行预设轮模型训练,得到目标船舶类别分类模型。本发明中的目标船舶类别分类模型正确率高,错误率较少。
公开/授权文献
- CN114627338A 一种船舶类别分类模型训练方法、系统及计算机存储介质 公开/授权日:2022-06-14