[發明專利]一種圖像目標檢測方法、系統、電子設備及存儲介質

在審

• 申請號: 201911025107.0
• 申請日: 2019-10-25
• 公開（公告）號: CN110781819A
• 公開（公告）日: 2020-02-11
• 發明（設計）人: 劉海威;董剛;梁玲燕;楊宏斌;曹其春;趙雅倩
• 申請（專利權）人: 浪潮電子信息產業股份有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/46;G06N3/04
• 代理公司: 11227 北京集佳知識產權代理有限公司
• 代理人: 郄晨芳

1.一種圖像目標檢測方法，其特征在于，應用于異構平臺的并行處理芯片，所述圖像目標檢測方法包括：

接收卷積神經網絡對圖像進行目標識別得到的多個目標對象邊界框的邊界框信息；其中，所述邊界框信息包括邊界框位置信息；

將所述邊界框信息存儲至隨機存取存儲器RAM；

依次讀取所述隨機存取存儲器RAM中的所述邊界框信息中的邊界框位置信息，并對讀取的邊界框位置信息多路并行執行基于非極大值抑制算法的局部最大搜索操作，得到預設數量的待輸出邊界框；

將所述待輸出邊界框的邊界框信息傳輸至所述異構平臺的中央處理器CPU，以便所述中央處理器CPU輸出圖像目標檢測結果。

2.根據權利要求1所述圖像目標檢測方法，其特征在于，所述邊界框信息還包括邊界框分類概率信息；

相應的，將所述邊界框信息存儲至隨機存取存儲器RAM包括：

將所述邊界框信息按照所述邊界框分類概率信息從高到低的順序寫入所述隨機存取存儲器RAM。

3.根據權利要求2所述圖像目標檢測方法，其特征在于，所述對讀取的邊界框位置信息多路并行執行基于非極大值抑制算法的局部最大搜索操作，得到預設數量的待輸出邊界框包括：

確定所述并行處理芯片的并行處理路數K；

按照所述邊界框分類概率信息從高到低的順序依次讀取所述邊界框位置信息，得到K個基準邊界框位置信息；其中，任意兩個所述基準邊界框位置信息的重疊面積小于預設值；

根據K個所述基準邊界框位置信息對所有邊界框位置信息并行執行基于非極大值抑制算法的局部最大搜索操作，得到預設數量的所述待輸出邊界框。

4.根據權利要求3所述圖像目標檢測方法，其特征在于，還包括：

當目標邊界框位置信息對應的目標對象邊界框不為所述待輸出邊界框時，將所述目標邊界框位置信息從所述隨機存取存儲器RAM中刪除。

5.根據權利要求1所述圖像目標檢測方法，其特征在于，所述并行處理芯片為現場可編程邏輯門陣列FPGA、ASIC芯片或嵌入式芯片。

6.根據權利要求1所述圖像目標檢測方法，其特征在于，所述邊界框位置信息包括邊界框X軸方向坐標、邊界框Y軸方向坐標、邊界框寬度和邊界框高度。

7.根據權利要求1至6任一項所述圖像目標檢測方法，其特征在于，所述邊界框信息還包括每一目標對象邊界框的前景置信度；

相應的，將所述邊界框信息存儲至隨機存取存儲器RAM包括：

將所述前景置信度大于或等于預設值的邊界框信息存儲至所述隨機存取存儲器RAM。

8.一種圖像目標檢測系統，其特征在于，應用于異構平臺的并行處理芯片，所述圖像目標檢測系統包括：

信息接收模塊，用于接收卷積神經網絡對圖像進行目標識別得到的多個目標對象邊界框的邊界框信息；其中，所述邊界框信息包括邊界框位置信息；

信息存儲模塊，用于將所述邊界框信息存儲至隨機存取存儲器RAM；

邊界框選擇模塊，用于依次讀取所述隨機存取存儲器RAM中的所述邊界框信息中的邊界框位置信息，并對讀取的邊界框位置信息多路并行執行基于非極大值抑制算法的局部最大搜索操作，得到預設數量的待輸出邊界框；

結果輸出模塊，用于將所述待輸出邊界框的邊界框信息傳輸至所述異構平臺的中央處理器CPU，以便所述中央處理器CPU輸出圖像目標檢測結果。

9.一種電子設備，其特征在于，包括存儲器和處理器，所述存儲器中存儲有計算機程序，所述處理器調用所述存儲器中的計算機程序時實現如權利要求1至7任一項所述圖像目標檢測方法的步驟。

10.一種存儲介質，其特征在于，所述存儲介質中存儲有計算機可執行指令，所述計算機可執行指令被處理器加載并執行時，實現如上權利要求1至7任一項所述圖像目標檢測方法的步驟。