[發(fā)明專利]圖像校正方法、圖像校正裝置和電子設(shè)備

在審

• 申請(qǐng)?zhí)枺?/td>: 201811208112.0
• 申請(qǐng)日: 2018-10-17
• 公開（公告）號(hào): CN111064860A
• 公開（公告）日: 2020-04-24
• 發(fā)明（設(shè)計(jì)）人: 肖雪;許譯天
• 申請(qǐng)（專利權(quán)）人: 北京地平線機(jī)器人技術(shù)研發(fā)有限公司
• 主分類號(hào): H04N1/60
• 分類號(hào): H04N1/60
• 代理公司: 北京市正見永申律師事務(wù)所 11497
• 代理人: 黃小臨;馮玉清

1.一種圖像校正方法，包括：

接收待校正圖像；以及

用色彩校正模型對(duì)所述待校正圖像進(jìn)行色彩校正，

其中，所述色彩校正模型是利用不同色溫下色卡的圖像訓(xùn)練得到的。

2.如權(quán)利要求1所述的方法，其中，所述色彩校正模型是通過(guò)如下步驟訓(xùn)練得到的：

獲取不同色溫下色卡的圖像；

從所述不同色溫下色卡的圖像確定所述色卡的與所述不同色溫對(duì)應(yīng)的圖像原色值；以及

用所述色卡的圖像原色值和對(duì)應(yīng)的色溫值作為輸入數(shù)據(jù)，以所述色卡的標(biāo)準(zhǔn)原色值作為監(jiān)督數(shù)據(jù)，來(lái)訓(xùn)練神經(jīng)網(wǎng)絡(luò)，以確定所述色彩校正模型。

3.如權(quán)利要求2所述的方法，其中，用色彩校正模型對(duì)所述待校正圖像進(jìn)行色彩校正包括：

以所述待校正圖像的圖像原色值和色溫值作為所述色彩校正模型的輸入，由所述色彩校正模型輸出校正后的原色值。

4.如權(quán)利要求1所述的方法，其中，所述色彩校正模型是通過(guò)如下步驟訓(xùn)練得到的：

獲取不同色溫下色卡的圖像；

從所述不同色溫下色卡的圖像確定所述色卡的與所述不同色溫對(duì)應(yīng)的圖像原色值；以及

分別用所述色卡在每一個(gè)色溫下的圖像原色值作為輸入數(shù)據(jù)，以所述色卡的標(biāo)準(zhǔn)原色值作為監(jiān)督數(shù)據(jù)，來(lái)訓(xùn)練神經(jīng)網(wǎng)絡(luò)，以確定在該色溫下的色彩校正模型。

5.如權(quán)利要求4所述的方法，還包括：

基于所述待校正圖像的色溫選擇與所述待校正圖像對(duì)應(yīng)的色彩校正模型，

其中，用色彩校正模型對(duì)所述待校正圖像進(jìn)行色彩校正包括以所述待校正圖像的圖像原色值作為所述色彩校正模型的輸入，由所述色彩校正模型輸出校正后的原色值。

6.如權(quán)利要求2至5中的任一項(xiàng)所述的方法，其中，所述色卡的圖像和所述待校正圖像二者都是原始(RAW)圖像或原色圖像。

7.如權(quán)利要求1所述的方法，其中，所述色彩校正模型是通過(guò)如下步驟訓(xùn)練得到的：

獲取不同色溫下色卡的原始(RAW)圖像；

對(duì)所述不同色溫下色卡的原始(RAW)圖像進(jìn)行白平衡歸一化處理；

從白平衡歸一化處理后的色卡的原始(RAW)圖像確定所述色卡的圖像原色值；以及

用所述色卡的圖像原色值作為輸入數(shù)據(jù)，以所述色卡的標(biāo)準(zhǔn)原色值作為監(jiān)督數(shù)據(jù)，來(lái)訓(xùn)練神經(jīng)網(wǎng)絡(luò)，以確定所述色彩校正模型。

8.如權(quán)利要求7所述的方法，其中，當(dāng)所述待校正圖像是原始(RAW)圖像時(shí)，所述方法還包括在對(duì)所述待校正圖像進(jìn)行色彩校正之前，對(duì)所述待校正圖像進(jìn)行白平衡歸一化處理。

9.如權(quán)利要求2、3和7中的任一項(xiàng)所述的方法，其中，所述色卡的圖像原色值和標(biāo)準(zhǔn)原色值被歸一化至[0,1]的范圍以對(duì)所述神經(jīng)網(wǎng)絡(luò)進(jìn)行訓(xùn)練，且

其中，所述待校正圖像的圖像原色值被歸一化至[0,1]的范圍，以輸入到所述色彩校正模型中進(jìn)行校正，校正后的圖像原色值被恢復(fù)到原始范圍。

10.如權(quán)利要求2、3和7中的任一項(xiàng)所述的方法，其中，所述待校正圖像的圖像原色值的位數(shù)與用于訓(xùn)練神經(jīng)網(wǎng)絡(luò)的圖像原色值的位數(shù)相同。

11.如權(quán)利要求2、3和7中的任一項(xiàng)所述的方法，其中，所述神經(jīng)網(wǎng)絡(luò)是反向傳播神經(jīng)網(wǎng)絡(luò)。

12.一種圖像校正裝置，包括：

訓(xùn)練模塊，用于利用不同色溫下色卡的圖像訓(xùn)練神經(jīng)網(wǎng)絡(luò)，以確定色彩校正模型；

接收模塊，用于接收待校正圖像；以及

校正模塊，用于用所述色彩校正模型對(duì)所述待校正圖像進(jìn)行色彩校正。

13.如權(quán)利要求12所述的裝置，還包括：

校正模型選擇模塊，用于基于所述待校正圖像的色溫，選擇與所述待校正圖像對(duì)應(yīng)的色彩校正模型。