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

基本信息

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

摘要

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