[發(fā)明專利]圖像處理模型的訓(xùn)練方法和裝置、白平衡處理方法和裝置在審
| 申請?zhí)枺?/td> | 202110317455.6 | 申請日: | 2021-03-25 |
| 公開(公告)號: | CN112884693A | 公開(公告)日: | 2021-06-01 |
| 發(fā)明(設(shè)計)人: | 張英琦 | 申請(專利權(quán))人: | 維沃移動通信(深圳)有限公司 |
| 主分類號: | G06T5/50 | 分類號: | G06T5/50;G06T5/00;G06N3/08;G06N3/04 |
| 代理公司: | 北京友聯(lián)知識產(chǎn)權(quán)代理事務(wù)所(普通合伙) 11343 | 代理人: | 尚志峰;汪海屏 |
| 地址: | 518101 廣東省深圳市寶安區(qū)新安街道海旺社區(qū)N12區(qū)新湖路99號*** | 國省代碼: | 廣東;44 |
| 權(quán)利要求書: | 查看更多 | 說明書: | 查看更多 |
| 摘要: | |||
| 搜索關(guān)鍵詞: | 圖像 處理 模型 訓(xùn)練 方法 裝置 白平衡 | ||
1.一種圖像處理模型的訓(xùn)練方法,其特征在于,包括:
獲取第一訓(xùn)練集圖像;
對所述第一訓(xùn)練集圖像進行偏色處理,得到偏色后的第二訓(xùn)練集圖像;
通過所述第一訓(xùn)練集圖像和所述第二訓(xùn)練集圖像對待訓(xùn)練神經(jīng)網(wǎng)絡(luò)進行訓(xùn)練,得到用于白平衡處理的所述圖像處理模型。
2.根據(jù)權(quán)利要求1所述的圖像處理模型的訓(xùn)練方法,其特征在于,所述待訓(xùn)練神經(jīng)網(wǎng)絡(luò)包括編碼器網(wǎng)絡(luò),所述對所述第一訓(xùn)練集圖像進行偏色處理,得到第二訓(xùn)練集圖像,包括:
通過所述編碼器網(wǎng)絡(luò)對所述第一訓(xùn)練集圖像進行卷積處理,得到第一特征圖數(shù)據(jù);
獲取預(yù)設(shè)的偏色數(shù)據(jù)和激活函數(shù);
通過所述激活函數(shù)將所述偏色數(shù)據(jù)添加至所述第一特征圖數(shù)據(jù)中,得到偏色的第二特征圖數(shù)據(jù);
根據(jù)所述第二特征圖數(shù)據(jù)生成所述第二訓(xùn)練集圖像。
3.根據(jù)權(quán)利要求2所述的圖像處理模型的訓(xùn)練方法,其特征在于,所述待訓(xùn)練神經(jīng)網(wǎng)絡(luò)還包括解碼器網(wǎng)絡(luò),所述通過所述第一訓(xùn)練集圖像和所述第二訓(xùn)練集圖像對待訓(xùn)練神經(jīng)網(wǎng)絡(luò)進行訓(xùn)練,包括:
通過所述解碼器網(wǎng)絡(luò)對所述第二訓(xùn)練集圖像進行反卷積處理,得到輸出圖像;
獲取預(yù)設(shè)的損失函數(shù),將所述輸出圖像和所述第一訓(xùn)練集圖像輸入至所述損失函數(shù);
獲取所述損失函數(shù)輸出的當(dāng)前損失值,以目標(biāo)損失值范圍為目標(biāo)持續(xù)訓(xùn)練所述待訓(xùn)練神經(jīng)網(wǎng)絡(luò),直至所述當(dāng)前損失值落入所述目標(biāo)損失值范圍內(nèi)。
4.根據(jù)權(quán)利要求3所述的圖像處理模型的訓(xùn)練方法,其特征在于,所述激活函數(shù)為線性整流函數(shù),所述損失函數(shù)為基于Wasserstein距離的損失函數(shù)。
5.根據(jù)權(quán)利要求3或4所述的圖像處理模型的訓(xùn)練方法,其特征在于,所述損失函數(shù)為:
其中,X為輸入變量,Z為編碼數(shù)據(jù),ObjWCAE(PX,PY)為所述當(dāng)前損失值,為X的分布的期望值,為Z關(guān)于X的條件概率分布的期望值,λ為正則項的系數(shù)且λ為常數(shù),為編碼器權(quán)重的L2范數(shù),為解碼器權(quán)重的L2范數(shù),β為稀疏項的系數(shù)且β為常數(shù),||h||1為稀疏項的L1范數(shù)。
6.根據(jù)權(quán)利要求2至4中任一項所述的圖像處理模型的訓(xùn)練方法,其特征在于,在所述通過所述第一訓(xùn)練集圖像和所述第二訓(xùn)練集圖像對待訓(xùn)練神經(jīng)網(wǎng)絡(luò)進行訓(xùn)練之后,所述訓(xùn)練方法還包括:
固定所述第二特征圖數(shù)據(jù)對應(yīng)的偏色權(quán)重值。
7.一種圖像處理模型的訓(xùn)練裝置,其特征在于,包括:
第一獲取單元,用于獲取第一訓(xùn)練集圖像;
第一處理單元,用于對所述第一訓(xùn)練集圖像進行偏色處理,得到偏色后的第二訓(xùn)練集圖像;
訓(xùn)練單元,通過所述第一訓(xùn)練集圖像和所述第二訓(xùn)練集圖像對待訓(xùn)練神經(jīng)網(wǎng)絡(luò)進行訓(xùn)練,得到用于白平衡處理的所述圖像處理模型。
8.一種圖像的白平衡處理方法,其特征在于,包括:
獲取根據(jù)權(quán)利要求1至6中任一項所述的圖像處理模型的訓(xùn)練方法訓(xùn)練得到的所述圖像處理模型;
通過所述圖像處理模型對待處理圖像進行白平衡處理,得到白平衡處理后的目標(biāo)圖像。
9.根據(jù)權(quán)利要求8所述的圖像的白平衡處理方法,其特征在于,所述圖像處理模型包括編碼器網(wǎng)絡(luò)和解碼器網(wǎng)絡(luò),所述通過所述圖像處理模型對待處理圖像進行白平衡處理,包括:
獲取偏色權(quán)重值;
通過所述編碼器網(wǎng)絡(luò)對所述待處理圖像進行卷積處理,得到所述待處理圖像對應(yīng)的第三特征圖數(shù)據(jù);
通過所述偏色權(quán)重值對所述第三特征圖數(shù)據(jù)進行加權(quán)操作,得到第四特征圖數(shù)據(jù);
通過所述解碼器網(wǎng)絡(luò)對所述第四特征圖數(shù)據(jù)進行反卷積處理,得到白平衡處理后所述目標(biāo)圖像。
該專利技術(shù)資料僅供研究查看技術(shù)是否侵權(quán)等信息,商用須獲得專利權(quán)人授權(quán)。該專利全部權(quán)利屬于維沃移動通信(深圳)有限公司,未經(jīng)維沃移動通信(深圳)有限公司許可,擅自商用是侵權(quán)行為。如果您想購買此專利、獲得商業(yè)授權(quán)和技術(shù)合作,請聯(lián)系【客服】
本文鏈接:http://www.17sss.com.cn/pat/books/202110317455.6/1.html,轉(zhuǎn)載請聲明來源鉆瓜專利網(wǎng)。
- 上一篇:一種部分流泵的單級套件及其多級部分流泵
- 下一篇:一種減振寬邊檢查井蓋
- 彩色圖像和單色圖像的圖像處理
- 圖像編碼/圖像解碼方法以及圖像編碼/圖像解碼裝置
- 圖像處理裝置、圖像形成裝置、圖像讀取裝置、圖像處理方法
- 圖像解密方法、圖像加密方法、圖像解密裝置、圖像加密裝置、圖像解密程序以及圖像加密程序
- 圖像解密方法、圖像加密方法、圖像解密裝置、圖像加密裝置、圖像解密程序以及圖像加密程序
- 圖像編碼方法、圖像解碼方法、圖像編碼裝置、圖像解碼裝置、圖像編碼程序以及圖像解碼程序
- 圖像編碼方法、圖像解碼方法、圖像編碼裝置、圖像解碼裝置、圖像編碼程序、以及圖像解碼程序
- 圖像形成設(shè)備、圖像形成系統(tǒng)和圖像形成方法
- 圖像編碼裝置、圖像編碼方法、圖像編碼程序、圖像解碼裝置、圖像解碼方法及圖像解碼程序
- 圖像編碼裝置、圖像編碼方法、圖像編碼程序、圖像解碼裝置、圖像解碼方法及圖像解碼程序





