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

基本信息

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

摘要

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