[發(fā)明專利]基于注意力機制的道路圖像目標(biāo)檢測方法及相關(guān)設(shè)備

在審

• 申請?zhí)枺?/td>: 202010936332.6
• 申請日: 2020-09-08
• 公開（公告）號: CN112101169A
• 公開（公告）日: 2020-12-18
• 發(fā)明（設(shè)計）人: 劉杰;王健宗;瞿曉陽
• 申請（專利權(quán)）人: 平安科技（深圳）有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/46;G06K9/62;G06N3/04;G06N3/08;G06N20/00
• 代理公司: 深圳市世聯(lián)合知識產(chǎn)權(quán)代理有限公司 44385
• 代理人: 汪琳琳

本申請實施例屬于機器學(xué)習(xí)技術(shù)領(lǐng)域，應(yīng)用于智慧城市領(lǐng)域中，涉及一種基于注意力機制的道路圖像目標(biāo)檢測方法，包括獲取待檢測道路圖像；將所述待檢測道路圖像輸入到訓(xùn)練好的檢測集成框架中，以及與各目標(biāo)檢測框架對應(yīng)的注意力權(quán)重值；通過各目標(biāo)檢測框架對所述待檢測道路圖像進行目標(biāo)檢測處理，輸出得到行人檢測結(jié)果；根據(jù)所述注意力權(quán)重值綜合各所述行人檢測結(jié)果，得到所述檢測集成框架的目標(biāo)檢測結(jié)果。本申請還提供一種基于注意力機制的道路圖像目標(biāo)檢測裝置、計算機設(shè)備及存儲介質(zhì)。此外，本申請還涉及區(qū)塊鏈技術(shù)，采用本方法能夠解決現(xiàn)有技術(shù)中的目標(biāo)檢測框架難以實現(xiàn)以針對性的設(shè)定的指標(biāo)來進行目標(biāo)檢測的技術(shù)問題。

技術(shù)領(lǐng)域

本申請涉及機器學(xué)習(xí)技術(shù)領(lǐng)域，特別是涉及一種基于注意力機制的道路圖像目標(biāo)檢測方法、裝置、計算機設(shè)備和存儲介質(zhì)。

背景技術(shù)

隨著近年來深度學(xué)習(xí)的迅猛發(fā)展，計算機視覺領(lǐng)域也取得了較大突破。而目標(biāo)檢測作為計算機視覺領(lǐng)域核心任務(wù)之一，也在深度學(xué)習(xí)的加持下取得了較傳統(tǒng)方法極為明顯的進步。目標(biāo)檢測的任務(wù)是找出圖像或者視頻中感興趣物體，同時檢測出他們的位置、大小和類別。現(xiàn)有技術(shù)中的Yolo系列和SSD系列模型會采用one stage算法實現(xiàn)分類與回歸，其僅僅使用一個CNN網(wǎng)絡(luò)直接預(yù)測不同目標(biāo)的類別和位置，這樣做雖然提高了模型的檢測速度，但是在精度上損失卻很大。目標(biāo)檢測的性能評價指標(biāo)也可以很具任務(wù)的不同而選擇不同的指標(biāo)，包括精確率、精確度、召回率、mean Average Precision(mAP)、intersectionover-union(IoU)等等，不同的任務(wù)往往需要不同的指標(biāo)。與此同時，AutoML技術(shù)也在目標(biāo)檢測領(lǐng)域得到了應(yīng)用，例如，MobileNetV3中使用了MnasNet和NetAdapt來對網(wǎng)絡(luò)進行粗略的搜索和微調(diào)。可以說，目標(biāo)檢測領(lǐng)域有著繁多的算法和指標(biāo)，各個模型的性能可能在不同的任務(wù)上有著較大的差異，而不同的任務(wù)也需要使用不同的指標(biāo)來對模型的性能進行定義，因此，在以特定的任務(wù)上，目標(biāo)檢測框架難以實現(xiàn)以針對性的設(shè)定的指標(biāo)來進行目標(biāo)檢測的目的。

發(fā)明內(nèi)容

基于此，針對上述技術(shù)問題，本申請?zhí)峁┮环N基于注意力機制的道路圖像目標(biāo)檢測方法、裝置、計算機設(shè)備及存儲介質(zhì)，以解決現(xiàn)有技術(shù)中的目標(biāo)檢測框架難以實現(xiàn)以針對性的設(shè)定的指標(biāo)來進行目標(biāo)檢測的技術(shù)問題。

一種基于注意力機制的道路圖像目標(biāo)檢測方法，所述方法包括：

獲取待檢測道路圖像；

將所述待檢測道路圖像輸入到訓(xùn)練好的檢測集成框架中，其中，所述檢測集成框架中包括至少兩個目標(biāo)檢測框架，以及與各目標(biāo)檢測框架對應(yīng)的注意力權(quán)重值；

通過各目標(biāo)檢測框架對所述待檢測道路圖像進行目標(biāo)檢測處理，輸出得到行人檢測結(jié)果；

根據(jù)所述注意力權(quán)重值綜合各所述行人檢測結(jié)果，得到所述檢測集成框架的目標(biāo)檢測結(jié)果。

一種基于注意力機制的道路圖像目標(biāo)檢測裝置，所述裝置包括：

數(shù)據(jù)模塊，用于獲取待檢測道路圖像；

輸入模塊，用于將所述待檢測道路圖像輸入到訓(xùn)練好的檢測集成框架中，其中，所述檢測集成框架中包括至少兩個目標(biāo)檢測框架，以及與各目標(biāo)檢測框架對應(yīng)的注意力權(quán)重值；

檢測模塊，用于通過各目標(biāo)檢測框架對所述待檢測道路圖像進行目標(biāo)檢測處理，輸出得到行人檢測結(jié)果；

綜合模塊，用于根據(jù)所述注意力權(quán)重值綜合各所述行人檢測結(jié)果，得到所述檢測集成框架的目標(biāo)檢測結(jié)果。

一種計算機設(shè)備，包括存儲器和處理器，以及存儲在所述存儲器中并可在所述處理器上運行的計算機可讀指令，所述處理器執(zhí)行所述計算機可讀指令時實現(xiàn)上述基于注意力機制的道路圖像目標(biāo)檢測方法的步驟。