一種使用神經(jīng)網(wǎng)絡(luò)進(jìn)行圖像白平衡校正的方法及裝置

基本信息

申請(qǐng)?zhí)?/td> CN201710826612.X 申請(qǐng)日 -
公開（公告）號(hào) CN107578390B 公開（公告）日 2020-08-07
申請(qǐng)公布號(hào) CN107578390B 申請(qǐng)公布日 2020-08-07
分類號(hào) G06T5/00;G06N3/04;G06N3/08 分類 -
發(fā)明人 劉娟;許會(huì);張智福;余思洋 申請(qǐng)（專利權(quán)）人 長(zhǎng)沙全度影像科技有限公司
代理機(jī)構(gòu) - 代理人 -
地址 410205 湖南省長(zhǎng)沙市高新開發(fā)區(qū)尖山路39號(hào)中電軟件園一期9棟廠房4層402房
法律狀態(tài) -

摘要

摘要 一種使用神經(jīng)網(wǎng)絡(luò)進(jìn)行圖像白平衡校正的方法及裝置，屬于數(shù)字圖像處理技術(shù)領(lǐng)域。包括：S1、采集原始圖像；S2、環(huán)境光源色溫的參數(shù)值計(jì)算；S3、對(duì)所述原始圖像和所述待處理圖像分別進(jìn)行預(yù)處理得到訓(xùn)練用的預(yù)處理圖像和第二預(yù)處理圖像；S4、構(gòu)建卷積神經(jīng)網(wǎng)絡(luò)模型；S5、訓(xùn)練卷積神經(jīng)網(wǎng)絡(luò)模型；S6、利用卷積神經(jīng)網(wǎng)絡(luò)計(jì)算得到紅色通道增益值gainR及藍(lán)色通道增益值gainB；S7、利用所述紅色通道增益值gainR及藍(lán)色通道增益值gainB對(duì)待處理圖像進(jìn)行白平衡校正，從而得到校正后的白平衡圖像。該方法解決了現(xiàn)有技術(shù)中白平衡容易失效的技術(shù)問題，有效的提高了算法的計(jì)算速度，精確度得到很大提高，且該模型魯棒性非常好。