[發明專利]圖像增強模型的訓練方法及裝置、圖像增強方法及裝置

有效

• 申請號: 202011475833.5
• 申請日: 2020-12-14
• 公開（公告）號: CN113538304B
• 公開（公告）日: 2023-08-18
• 發明（設計）人: 劉恩雨;李松南
• 申請（專利權）人: 騰訊科技（深圳）有限公司
• 主分類號: G06T5/50
• 分類號: G06T5/50;G06T7/90;G06N3/0464;G06N3/092;G06N3/096
• 代理公司: 北京勵誠知識產權代理有限公司 11647
• 代理人: 趙爽

1.一種圖像增強模型的訓練方法，其特征在于，包括：

獲取訓練圖像集，所述訓練圖像集中的各個訓練樣本包括輸入圖像以及配對的目標圖像，其中，所述目標圖像是由所述輸入圖像經色彩增強處理得到的增強圖像與所述輸入圖像對應的HDR圖像融合后得到的圖像；

根據所述訓練圖像集對初始圖像增強模型進行模型訓練，得到目標圖像增強模型；

所述方法還包括：

將所述輸入圖像的RGB三通道分離，得到R通道、G通道和B通道；

根據所述R通道、所述G通道和所述B通道中每個顏色通道的顏色直方圖和目標裁剪步長，分別對所述R通道、所述G通道和所述B通道中每個顏色通道的像素值進行修正，得到所述增強圖像；

所述方法還包括：

獲取所述輸入圖像的初始裁剪步長；

根據所述輸入圖像的寬度和高度、所述初始裁剪步長，確定裁剪像素點的數量n₀；

根據所述裁剪像素點的數量n₀，對所述R通道、所述G通道和所述B通道中每個顏色通道的顏色直方圖進行裁剪，分別得到所述R通道的下邊界R_下、所述G通道的下邊界G_下和所述B通道的下邊界B_下，以及所述R通道的上邊界R_上、所述G通道的上邊界G_上和所述B通道的上邊界B_上；

根據所述R通道的下邊界R_下和上邊界R_上、所述G通道的下邊界G_下和上邊界G_上、所述B通道的下邊界B_下和上邊界B_上和所述初始裁剪步長，確定所述目標裁剪步長。

2.根據權利要求1所述的方法，其特征在于，所述根據所述裁剪像素點的數量n₀，對所述R通道、所述G通道和所述B通道中每個顏色通道的顏色直方圖進行裁剪，分別得到所述R通道的下邊界R_下、所述G通道的下邊界G_下和所述B通道的下邊界B_下，以及所述R通道的上邊界R_上、所述G通道的上邊界G_上和所述B通道的上邊界B_上，包括：

將所述R通道、所述G通道和所述B通道中每個顏色通道上初始分量由小到大的順序確定為第一裁剪順序，以及將所述R通道、所述G通道和所述B通道中每個顏色通道上初始分量由大到小的順序確定為第二裁剪順序；

根據所述裁剪像素點的數量n₀和所述第一裁剪順序，對所述R通道、所述G通道和所述B通道中每個顏色通道的顏色直方圖進行裁剪，分別得到所述R通道的下邊界R_下、所述G通道的下邊界G_下和所述B通道的下邊界B_下；

根據所述裁剪像素點的數量n₀和所述第二裁剪順序，對所述R通道、所述G通道和所述B通道中每個顏色通道的顏色直方圖進行裁剪，分別得到所述R通道的上邊界R_上、所述G通道的上邊界G_上和所述B通道的上邊界B_上。