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

有效

• 申請號: 201910576414.1
• 申請日: 2019-06-28
• 公開（公告）號: CN110276767B
• 公開（公告）日: 2021-08-31
• 發明（設計）人: 黃海東
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06T7/11
• 分類號: G06T7/11;G06T7/194;G06T7/66
• 代理公司: 廣州華進聯合專利商標代理有限公司 44224
• 代理人: 方高明

本申請涉及一種圖像處理方法、裝置、電子設備和計算機可讀存儲介質。所述方法包括獲取可見光圖；生成與所述可見光圖對應的中心權重圖，其中，所述中心權重圖所表示的權重值從中心到邊緣逐漸減小；將所述可見光圖和所述中心權重圖輸入到主體檢測模型中，得到主體區域置信度圖，其中，所述主體檢測模型是預先根據同一場景的可見光圖、中心權重圖及對應的已標注的主體掩膜圖進行訓練得到的模型；根據所述主體區域置信度圖確定所述可見光圖中的目標主體。能夠準確的識別圖像中的目標主體。

技術領域

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

背景技術

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

發明內容

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

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

獲取可見光圖；

生成與所述可見光圖對應的中心權重圖，其中，所述中心權重圖所表示的權重值從中心到邊緣逐漸減小；

將所述可見光圖和所述中心權重圖輸入到主體檢測模型中，得到主體區域置信度圖，其中，所述主體檢測模型是預先根據同一場景的可見光圖、中心權重圖及對應的已標注的主體掩膜圖進行訓練得到的模型；

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

一種主體檢測模型的訓練方法，包括：

獲取同一場景的可見光圖、深度圖和已標注的主體掩膜圖；

生成與所述可見光圖對應的中心權重圖，其中，所述中心權重圖所表示的權重值從中心到邊緣逐漸減小；

將所述可見光圖、所述深度圖和所述中心權重圖作為包含初始網絡權重的主體檢測模型的輸入，將所述已標注的主體掩膜圖作為包含初始網絡權重的主體檢測模型輸出的真實值，進行訓練，得到主體檢測模型的目標網絡權重。

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

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

權重生成模塊，用于生成與所述可見光圖對應的中心權重圖，其中，所述中心權重圖所表示的權重值從中心到邊緣逐漸減小；

檢測模塊，用于將所述可見光圖和所述中心權重圖輸入到主體檢測模型中，得到主體區域置信度圖，其中，所述主體檢測模型是預先根據同一場景的可見光圖、中心權重圖及對應的已標注的主體掩膜圖進行訓練得到的模型；

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

一種主體檢測模型的訓練裝置，包括：

訓練圖像獲取模塊，用于獲取同一場景的可見光圖、深度圖和已標注的主體掩膜圖；

訓練權重生成模塊，用于生成與所述可見光圖對應的中心權重圖，其中，所述中心權重圖所表示的權重值從中心到邊緣逐漸減小；

訓練模塊，用于將所述可見光圖、所述深度圖和所述中心權重圖作為包含初始網絡權重的主體檢測模型的輸入，將所述已標注的主體掩膜圖作為包含初始網絡權重的主體檢測模型輸出的真實值，進行訓練，得到主體檢測模型的目標網絡權重。

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