一種通用的嵌入式平臺的行人檢測模型訓(xùn)練方法

基本信息

申請?zhí)?/td> CN202110883375.7 申請日 -
公開(公告)號 CN113343949A 公開(公告)日 2021-09-03
申請公布號 CN113343949A 申請公布日 2021-09-03
分類號 G06K9/00;G06K9/62;G06T1/20;G06N3/08 分類 計(jì)算;推算;計(jì)數(shù);
發(fā)明人 王建興;王縱馳;劉健;臧媛媛;王培;于佳耕 申請(專利權(quán))人 航天神舟智慧系統(tǒng)技術(shù)有限公司
代理機(jī)構(gòu) 北京中知法苑知識產(chǎn)權(quán)代理有限公司 代理人 李明;趙吉陽
地址 100088 北京市海淀區(qū)馬甸路2號中航油大廈
法律狀態(tài) -

摘要

摘要 一種通用的基于嵌入式平臺的行人檢測模型的訓(xùn)練方法,包括:數(shù)據(jù)集的準(zhǔn)備,進(jìn)行圖像標(biāo)準(zhǔn)化;訓(xùn)練浮點(diǎn)模型,采用降低浮點(diǎn)數(shù)進(jìn)行深度學(xué)習(xí)模型訓(xùn)練;微調(diào)量化卷積層,使用上一步驟得到的浮點(diǎn)模型,并將模型的路徑添加到微調(diào)量化卷積腳本中進(jìn)行微調(diào),得到微調(diào)量化卷積層;應(yīng)用線性變換微調(diào)模型,利用已標(biāo)注的樣本對預(yù)訓(xùn)練參數(shù)進(jìn)行調(diào)整,以針對行人檢測任務(wù)來修正模型網(wǎng)絡(luò);模型效果測試,利用測試集測試行人檢測模型的準(zhǔn)確率和召回率;在實(shí)時性和檢測的靈活性上有較大的提高,同時具有高準(zhǔn)確率和召回率等良好性能。