[發明專利]圖像處理模型的訓練方法和裝置、白平衡處理方法和裝置

在審

• 申請號: 202110317455.6
• 申請日: 2021-03-25
• 公開（公告）號: CN112884693A
• 公開（公告）日: 2021-06-01
• 發明（設計）人: 張英琦
• 申請（專利權）人: 維沃移動通信(深圳)有限公司
• 主分類號: G06T5/50
• 分類號: G06T5/50;G06T5/00;G06N3/08;G06N3/04
• 代理公司: 北京友聯知識產權代理事務所(普通合伙) 11343
• 代理人: 尚志峰;汪海屏

本申請公開了一種圖像處理模型的訓練方法和裝置、白平衡處理方法和裝置，其中，圖像處理模型的訓練方法包括：獲取第一訓練集圖像；對第一訓練集圖像進行偏色處理，得到偏色后的第二訓練集圖像；通過第一訓練集圖像和第二訓練集圖像對待訓練神經網絡進行訓練，得到用于白平衡處理的圖像處理模型。

技術領域

本申請屬于圖像處理技術領域，具體涉及一種圖像處理模型的訓練方法和裝置、白平衡處理方法和裝置。

背景技術

在相關技術中，對于相機等圖像獲取設備，由于硬件傳感器的局限，傳感器輸出的圖像與人眼觀看的圖像會存在一定的差別，導致拍攝圖像中物體顏色與人眼識別的被攝物體的實際顏色不同。

一般來說，需要對拍攝的照片源片進行白平衡處理，目前基于深度神經網絡的白平衡方法基于監督和半監督方法，而該方法在模型訓練過程中，需要大量包含真實光源注釋的訓練集數據，而真實光源注釋需要消耗大量的時間，造成訓練集獲取困難，最終降低訓練所得模型的準確性。

因此，目前亟需一種能夠不依賴真實光源注釋的訓練集數據的訓練方法。

發明內容

本申請實施例的目的是提供一種圖像處理模型的訓練方法和裝置、圖像的白平衡處理方法和裝置，能夠不依賴真實光源注釋的訓練集數據來對圖像處理模型進行高效訓練。

為了解決上述技術問題，本申請是這樣實現的：

第一方面，本申請實施例提供了一種圖像處理模型的訓練方法，包括：

獲取第一訓練集圖像；

對第一訓練集圖像進行偏色處理，得到偏色后的第二訓練集圖像；

通過第一訓練集圖像和第二訓練集圖像對待訓練神經網絡進行訓練，得到用于白平衡處理的圖像處理模型。

第二方面，本申請實施例提供了一種圖像處理模型的訓練裝置，包括：

第一獲取單元，用于獲取第一訓練集圖像；

第一處理單元，用于對第一訓練集圖像進行偏色處理，得到偏色后的第二訓練集圖像；

訓練單元，通過第一訓練集圖像和第二訓練集圖像對待訓練神經網絡進行訓練，得到用于白平衡處理的圖像處理模型。

第三方面，本申請實施例提供了一種圖像的白平衡處理方法，包括：

獲取根據第一方面中提供的圖像處理模型的訓練方法訓練得到的圖像處理模型；

通過圖像處理模型對待處理圖像進行白平衡處理，得到白平衡處理后的目標圖像。

第四方面，本申請實施例提供了一種圖像的白平衡處理裝置，包括：

第二獲取單元，用于獲取根據第一方面中提供的圖像處理模型的訓練方法訓練得到的圖像處理模型；

第二處理單元，用于通過圖像處理模型對待處理圖像進行白平衡處理，得到白平衡處理后的目標圖像。

第五方面，本申請實施例提供了一種電子設備，包括處理器，存儲器及存儲在存儲器上并可在處理器上運行的程序或指令，程序或指令被處理器執行時實現如第一方面中提供的圖像處理模型的訓練方法；和/或

如第三方面中提供的圖像的白平衡處理方法。

第六方面，本申請實施例提供了一種可讀存儲介質，可讀存儲介質上存儲程序或指令，程序或指令被處理器執行時實現如第一方面中提供的圖像處理模型的訓練方法；和/或

如第三方面中提供的圖像的白平衡處理方法。