多任務(wù)的行為識別模型訓(xùn)練方法、裝置、設(shè)備及存儲介質(zhì)

基本信息

申請?zhí)?/td> 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 分類 計算；推算；計數(shù);
發(fā)明人 蘇婧;蘇海昇;王棟梁;甘偉豪 申請（專利權(quán)）人 上海商湯智能科技有限公司
代理機構(gòu) 北京派特恩知識產(chǎn)權(quán)代理有限公司 代理人 崔曉嵐;張穎玲
地址 200233上海市徐匯區(qū)桂平路391號3號樓1605A室
法律狀態(tài) -

摘要

摘要 本申請實施例提供一種多任務(wù)的行為識別模型訓(xùn)練方法、裝置、設(shè)備及存儲介質(zhì)，其中，所述方法包括：獲取N個待識別任務(wù)的樣本數(shù)據(jù)集，其中，所述樣本數(shù)據(jù)集包括虛擬數(shù)據(jù)、攝像裝置的采集數(shù)據(jù)和互聯(lián)網(wǎng)的爬取數(shù)據(jù)至少之一，N為大于等于1的整數(shù)；利用所述N個待識別任務(wù)的樣本數(shù)據(jù)集訓(xùn)練搭建的行為識別模型，得到多任務(wù)的行為識別模型，其中，所述搭建的行為識別模型包括第一主干網(wǎng)絡(luò)和N個第一分類網(wǎng)絡(luò)，所述多任務(wù)的行為識別模型包括多任務(wù)主干網(wǎng)絡(luò)和N個第二分類網(wǎng)絡(luò)，所述多任務(wù)主干網(wǎng)絡(luò)是通過訓(xùn)練所述第一主干網(wǎng)絡(luò)而得到，所述N個第二分類網(wǎng)絡(luò)是與所述N個待識別任務(wù)一一對應(yīng)的，通過訓(xùn)練N個所述第一分類網(wǎng)絡(luò)得到。