[發明專利]一種模型訓練以及確定無人車運動策略方法及裝置有效
| 申請號: | 202110191748.4 | 申請日: | 2021-02-19 |
| 公開(公告)號: | CN112883871B | 公開(公告)日: | 2022-06-10 |
| 發明(設計)人: | 劉朋浩 | 申請(專利權)人: | 北京三快在線科技有限公司 |
| 主分類號: | G06V20/58 | 分類號: | G06V20/58;G06V40/10;G06V10/764;G06V10/774;G06V10/82;G06K9/62;G06N3/04;G06N3/08 |
| 代理公司: | 北京曼威知識產權代理有限公司 11709 | 代理人: | 方志煒 |
| 地址: | 100080 北京市海*** | 國省代碼: | 北京;11 |
| 權利要求書: | 查看更多 | 說明書: | 查看更多 |
| 摘要: | |||
| 搜索關鍵詞: | 一種 模型 訓練 以及 確定 無人 運動 策略 方法 裝置 | ||
1.一種檢測模型的訓練方法,其特征在于,所述方法包括:
根據歷史上采集到的各連續幀圖像,確定若干組連續兩幀圖像,將每組圖像,作為訓練樣本,所述訓練樣本中的前一幀作為第一圖像,后一幀作為第二圖像;
針對每個訓練樣本,將該訓練樣本包含的第二圖像中,預先標注的各障礙物實例的包圍框,作為該訓練樣本的標注;
將該訓練樣本輸入訓練完成的預處理模型,通過所述預處理模型的卷積層和分類層,確定所述第一圖像中各障礙物實例的包圍框,以及通過所述預處理模型的卷積層,確定所述第二圖像中的各障礙物實例的各待選包圍框,并將所述第一圖像中各障礙物實例的包圍框和所述第二圖像中的各障礙物實例的各待選包圍框作為預處理結果;
將該訓練樣本的預處理結果,輸入待訓練的檢測模型,確定所述第一圖像中各障礙物實例的包圍框的預測結果,并根據所述預測結果,更新所述第二圖像中的各障礙物實例的至少部分待選包圍框,作為后處理結果;
將該訓練樣本的后處理結果,輸入所述預處理模型的分類層,得到該訓練樣本包含的第二圖像中各障礙物實例的包圍框,作為檢測結果;
以該訓練樣本的檢測結果以及該訓練樣本的標注之間差異最小為優化目標,調整所述待訓練的檢測模型的模型參數,所述預處理模型以及所述檢測模型用于在確定無人車運動策略時,確定所述無人車環境圖像中的各障礙物實例。
2.如權利要求1所述的方法,其特征在于,所述預處理模型包括卷積層和分類層;
將該訓練樣本輸入訓練完成的預處理模型,通過所述預處理模型的卷積層和分類層,確定所述第一圖像中各障礙物實例的包圍框,以及通過所述預處理模型的卷積層,確定所述第二圖像中的各障礙物實例的各待選包圍框,并將所述第一圖像中各障礙物實例的包圍框和所述第二圖像中的各障礙物實例的各待選包圍框作為預處理結果,具體包括:
將該訓練樣本包含的所述第一圖像與所述第二圖像作為輸入,輸入到所述預處理模型的卷積層,確定所述第一圖像與第二圖像中各障礙物實例的各待選包圍框;
將所述第一圖像的各障礙物實例的各待選包圍框輸入到所述預處理模型的分類層,確定所述第一圖像中各障礙物實例的包圍框,其中,所述分類層用于針對每個障礙物實例,從該障礙物實例的各待選包圍框中確定該障礙物實例的包圍框;
將所述第一圖像中各障礙物實例的包圍框與所述第二圖像中各障礙物實例的各待選包圍框,作為預處理結果。
3.如權利要求1所述的方法,其特征在于,所述檢測模型包括預測層和匹配層;
將該訓練樣本的預處理結果,輸入待訓練的檢測模型,確定所述第一圖像中各障礙物實例的包圍框的預測結果,并根據所述預測結果,更新所述第二圖像中的各障礙物實例的至少部分待選包圍框,具體包括:
將所述預處理結果包含的所述第一圖像中各障礙物實例的包圍框輸入待訓練的檢測模型的預測層,得到所述第一圖像中各障礙物實例的包圍框的預測結果,所述預測結果表征預測得到的各障礙物實例的在采集所述第一圖像的下一時刻的位置;
將各障礙物實例的包圍框的預測結果與所述預處理結果包含的第二圖像中的各障礙物實例的各待選包圍框作為輸入,輸入到所述檢測模型的匹配層,得到匹配結果;
根據所述匹配結果以及所述預測結果包含的包圍框,更新所述第二圖像中的各障礙物實例的至少部分待選包圍框。
4.如權利要求3所述的方法,其特征在于,根據所述匹配結果以及所述預測結果包含的包圍框,更新所述第二圖像中的各障礙物實例的至少部分待選包圍框,具體包括:
根據所述匹配結果以及所述預測結果包含的包圍框,確定所述預測結果包含的包圍框和所述第二圖像的包圍框的匹配部分和不匹配部分;
根據所述預測結果包含的包圍框的匹配部分,對所述第二圖像的包圍框的匹配部分進行替換;
將所述預測結果包含的包圍框的不匹配部分,加入所述第二圖像的包圍框。
該專利技術資料僅供研究查看技術是否侵權等信息,商用須獲得專利權人授權。該專利全部權利屬于北京三快在線科技有限公司,未經北京三快在線科技有限公司許可,擅自商用是侵權行為。如果您想購買此專利、獲得商業授權和技術合作,請聯系【客服】
本文鏈接:http://www.17sss.com.cn/pat/books/202110191748.4/1.html,轉載請聲明來源鉆瓜專利網。





