一種基于全卷積神經(jīng)網(wǎng)絡(luò)和特征金字塔的顏色校正方法

基本信息

申請(qǐng)?zhí)?/td> CN201910659156.3 申請(qǐng)日 -
公開（公告）號(hào) CN110400275B 公開（公告）日 2021-06-22
申請(qǐng)公布號(hào) CN110400275B 申請(qǐng)公布日 2021-06-22
分類號(hào) G06T5/00 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 李得元;代超;何帆;周振 申請(qǐng)（專利權(quán)）人 中電健康云科技有限公司
代理機(jī)構(gòu) 成都弘毅天承知識(shí)產(chǎn)權(quán)代理有限公司 代理人 黃蓉蓉
地址 610000 四川省成都市雙流區(qū)東升街道成都芯谷產(chǎn)業(yè)園集中區(qū)內(nèi)
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種基于全卷積神經(jīng)網(wǎng)絡(luò)和特征金字塔的顏色校正方法，涉及圖像顏色校正技術(shù)領(lǐng)域，本發(fā)明依次對(duì)圖像進(jìn)行顏色校正、裁剪、數(shù)據(jù)增強(qiáng)處理，得到對(duì)應(yīng)的偏色圖片，由若干偏色圖片構(gòu)成訓(xùn)練數(shù)據(jù)集；利用訓(xùn)練數(shù)據(jù)集對(duì)預(yù)設(shè)全卷積神經(jīng)網(wǎng)絡(luò)和特征金字塔模型進(jìn)行訓(xùn)練，基于預(yù)設(shè)的損失函數(shù)和優(yōu)化器，對(duì)預(yù)設(shè)全卷積神經(jīng)網(wǎng)絡(luò)和特征金字塔模型的網(wǎng)絡(luò)參數(shù)進(jìn)行優(yōu)化調(diào)整，直至網(wǎng)絡(luò)收斂，輸出訓(xùn)練好的全卷積神經(jīng)網(wǎng)絡(luò)和特征金字塔模型；利用訓(xùn)練好的模型預(yù)估待校正圖像的光源信息，根據(jù)預(yù)估得到的光源信息對(duì)待校正圖像進(jìn)行顏色校正，本發(fā)明通過全卷積神經(jīng)網(wǎng)絡(luò)和特征金字塔預(yù)估光源信息，利用伽馬校正對(duì)圖像進(jìn)行顏色校正，確保顏色校正效果，并且普適性高。