[發明專利]物體檢測方法、網絡訓練方法、裝置和計算機存儲介質

有效

• 申請號: 201810143218.0
• 申請日: 2018-02-11
• 公開（公告）號: CN108388859B
• 公開（公告）日: 2022-04-15
• 發明（設計）人: 岳曉宇;曠章輝;張兆陽;陳振方;張偉
• 申請（專利權）人: 深圳市商湯科技有限公司
• 主分類號: G06V10/25
• 分類號: G06V10/25;G06V20/40;G06N3/08
• 代理公司: 北京派特恩知識產權代理有限公司 11270
• 代理人: 李梅香;張穎玲

1.一種基于深度學習的物體檢測方法，其特征在于，所述方法包括：

基于第一圖像數據和區域估計網絡，獲得第二圖像數據，所述第二圖像數據包括所述第一圖像數據中待檢測物體的估計區域；

將所述第一圖像數據和所述第二圖像數據輸入物體檢測網絡，以使所述物體檢測網絡基于所述估計區域對所述第一圖像數據進行檢測，獲得所述第一圖像數據中所述待檢測物體的檢測區域。

2.根據權利要求1所述的方法，其特征在于，所述方法還包括：

獲得用于網絡訓練的第四圖像數據，以及所述第四圖像數據中包括的標注區域對應的標注信息；所述標注區域表征所述第四圖像數據中物體所在區域；

采用包含有向卷積層的卷積神經網絡對攜帶有標注區域的所述第四圖像數據、所述標注區域對應的標注信息進行訓練，獲得物體檢測網絡，所述卷積神經網絡用于在訓練過程中對所述第四圖像數據的標注區域進行計算處理。

3.根據權利要求2所述的方法，其特征在于，所述采用包含有向卷積層的卷積神經網絡對攜帶有標注區域的所述第四圖像數據、所述標注區域對應的標注信息進行訓練，獲得物體檢測網絡，包括：

確定除所述標注區域以外的背景區域的部分背景區域；所述部分背景區域采用隨機方式從所述背景區域中確定；

將所述部分背景區域添加至所述標注區域，生成所述第四 圖像數據的更新標注區域；

基于有向卷積層分別識別所述第四圖像數據中的標注區域和更新標注區域，對所述標注區域中的圖像數據以及對應的標注信息進行訓練，以及對所述更新標注區域中的圖像數據以及對應的標注信息進行訓練，獲得物體檢測網絡。

4.根據權利要求1所述的方法，其特征在于，所述基于第一圖像數據和區域估計網絡，獲得第二圖像數據，包括：

對所述第一圖像數據進行卷積處理，獲得所述第一圖像數據對應的第一特征圖；

將所述特征圖輸入至少一個池化層，獲得與所述特征圖的尺度不同的至少一個第二特征圖；

對具有不同尺度的所述第一特征圖、所述至少一個第二特征圖分別進行卷積處理，對卷積處理后的所述第一特征圖、所述至少一個第二特征圖進行融合，輸出攜帶有標識待檢測物體的估計區域的第二圖像數據。

5.根據權利要求4所述的方法，其特征在于，所述輸出攜帶有標識待檢測物體的估計區域的第二圖像數據，包括：

對融合后的所述第一特征圖、所述至少一個第二特征圖進行二值化處理，輸出二值化表示的攜帶有標識待檢測物體的估計區域的第二圖像數據。

6.根據權利要求1所述的方法，其特征在于，所述將所述第一圖像數據和所述第二圖像數據輸入物體檢測網絡，以使所述物體檢測網絡基于所述估計區域對所述第一圖像數據進行檢測，包括：

將所述第一圖像數據和所述第二圖像數據輸入物體檢測網絡；所述物體檢測網絡包括有向卷積層；

基于所述有向卷積層識別所述第二圖像數據中的所述估計區域，確定所述第一圖像數據中、與所述估計區域對應的目標區域，對所述第一圖像數據中的所述目標區域進行卷積操作。

7.一種網絡訓練方法，其特征在于，所述方法包括：

獲得用于網絡訓練的第四圖像數據，以及所述第四圖像數據中標注區域對應的標注信息；所述標注區域表征所述第四圖像數據中檢測物體所在區域；

采用包含有向卷積層的卷積神經網絡對攜帶有標注區域的所述第四圖像數據、所述標注區域對應的標注信息進行訓練，獲得物體檢測網絡，所述卷積神經網絡用于在訓練過程中對所述第四圖像數據的標注區域進行計算處理。

8.根據權利要求7所述的方法，其特征在于，所述采用包含有向卷積層的卷積神經網絡對攜帶有標注區域的所述第四圖像數據、所述標注區域對應的標注信息進行訓練，獲得物體檢測網絡，包括：

基于有向卷積層識別所述第四圖像數據中的標注區域，對所述標注區域中的圖像數據以及對應的標注信息進行訓練，獲得物體檢測網絡。