首页/ 上海商汤智能科技有限公司/ 专利详情

多任务的行为识别模型训练方法、装置、设备及存储介质

基本信息

申请号 CN202110572866.X 申请日 -
公开（公告）号 CN113344048A 公开（公告）日 2021-09-03
申请公布号 CN113344048A 申请公布日 2021-09-03
分类号 G06K9/62（2006.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分类 计算；推算；计数;
发明人 苏婧;苏海昇;王栋梁;甘伟豪 申请（专利权）人 上海商汤智能科技有限公司
代理机构 北京派特恩知识产权代理有限公司 代理人 崔晓岚;张颖玲
地址 200233上海市徐汇区桂平路391号3号楼1605A室
法律状态 -

摘要

摘要 本申请实施例提供一种多任务的行为识别模型训练方法、装置、设备及存储介质，其中，所述方法包括：获取N个待识别任务的样本数据集，其中，所述样本数据集包括虚拟数据、摄像装置的采集数据和互联网的爬取数据至少之一，N为大于等于1的整数；利用所述N个待识别任务的样本数据集训练搭建的行为识别模型，得到多任务的行为识别模型，其中，所述搭建的行为识别模型包括第一主干网络和N个第一分类网络，所述多任务的行为识别模型包括多任务主干网络和N个第二分类网络，所述多任务主干网络是通过训练所述第一主干网络而得到，所述N个第二分类网络是与所述N个待识别任务一一对应的，通过训练N个所述第一分类网络得到。