[發明專利]圖像處理方法和裝置、電子設備、計算機可讀存儲介質

有效

• 申請號: 201910726785.3
• 申請日: 2019-08-07
• 公開（公告）號: CN110473185B
• 公開（公告）日: 2022-03-15
• 發明（設計）人: 黃海東
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T7/50;G06V10/25;G06V10/80
• 代理公司: 華進聯合專利商標代理有限公司 44224
• 代理人: 董慧

本申請涉及一種圖像處理方法和裝置、電子設備、計算機可讀存儲介質，該方法包括：獲取可見光圖；將所述可見光圖輸入主體檢測模型的主體識別層中，得到主體識別圖；其中，所述主體檢測模型是根據同一場景的預設條件進行訓練得到的模型；將所述可見光圖輸入所述主體檢測模型的深度預測層中，得到深度預測圖；融合所述主體識別圖和所述深度預測圖，得到主體區域置信度圖；根據所述主體區域置信度圖確定所述可見光圖中的目標主體。能夠準確的識別圖像中的目標主體。

技術領域

本申請涉及影像領域，特別是涉及一種圖像處理方法和裝置、電子設備、計算機可讀存儲介質。

背景技術

隨著影像技術的發展，人們越來越習慣通過電子設備上的攝像頭等圖像采集設備拍攝圖像或視頻，記錄各種信息。攝像頭在采集圖像過程中有時需要檢測到主體，傳統的主體檢測方式無法準確的檢測出圖像中的主體。

發明內容

本申請實施例提供一種圖像處理方法和裝置、電子設備、計算機可讀存儲介質，能夠提高主體檢測的準確性。

一種圖像處理方法，包括：

獲取可見光圖；

將所述可見光圖輸入主體檢測模型的主體識別層中，得到主體識別圖；其中，所述主體檢測模型是根據同一場景的預設條件進行訓練得到的模型；

將所述可見光圖輸入所述主體檢測模型的深度預測層中，得到深度預測圖；

融合所述主體識別圖和所述深度預測圖，得到主體區域置信度圖；

根據所述主體區域置信度圖確定所述可見光圖中的目標主體。

一種圖像處理裝置，包括：

獲取模塊，用于獲取可見光圖；

識別模塊，用于將所述可見光圖輸入主體檢測模型的主體識別層中，得到主體識別圖；其中，所述主體檢測模型是根據同一場景的預設條件進行訓練得到的模型；

預測模塊，用于將所述可見光圖輸入所述主體檢測模型的深度預測層中，得到深度預測圖；

融合模塊，用于融合所述主體識別圖和所述深度預測圖，得到主體區域置信度圖；

確定模塊，用于根據所述主體區域置信度圖確定所述可見光圖中的目標主體。

一種電子設備，包括存儲器及處理器，所述存儲器中儲存有計算機程序，所述計算機程序被所述處理器執行時，使得所述處理器執行所述圖像處理方法的步驟。

一種計算機可讀存儲介質，其上存儲有計算機程序，所述計算機程序被處理器執行時實現所述圖像處理方法的步驟。

上述圖像處理方法和裝置、電子設備、計算機可讀存儲介質，獲取可見光圖，將可見光圖輸入主體檢測模型的主體識別層中，可以得到主體識別圖，從而初步識別出可見光圖中的主體。將可見光圖輸入主體檢測模型的深度預測層中，可以得到可見光圖對應的深度圖。通過雙路網絡識別得到深度圖和主體識別圖，再融合主體識別圖和深度預測圖，得到主體區域置信度圖，根據主體區域置信度圖確定可見光圖中的目標主體，從而更加準確的識別出可見光圖中的目標主體。

附圖說明

為了更清楚地說明本申請實施例或現有技術中的技術方案，下面將對實施例或現有技術描述中所需要使用的附圖作簡單地介紹，顯而易見地，下面描述中的附圖僅僅是本申請的一些實施例，對于本領域普通技術人員來講，在不付出創造性勞動的前提下，還可以根據這些附圖獲得其他的附圖。

圖1為一個實施例中電子設備的內部結構框圖；

圖2為一個實施例中圖像處理方法的流程圖；

圖3為一個實施例中融合主體識別圖和深度預測圖，得到主體區域置信度圖的步驟的流程圖；