一種基于光照變換網(wǎng)絡(luò)的圖像自動增強(qiáng)方法

基本信息

申請?zhí)?/td> CN202011624087.1 申請日 -
公開（公告）號 CN112686823A 公開（公告）日 2021-04-20
申請公布號 CN112686823A 申請公布日 2021-04-20
分類號 G06N5/04(2006.01)I;G06T5/00(2006.01)I;G06K9/46(2006.01)I;G06N3/08(2006.01)I;G06K9/62(2006.01)I;G06N3/04(2006.01)I 分類 計算；推算；計數(shù);
發(fā)明人 蘇家儀;韋光亮;王筱東;張玉國;蘇世寧;韋瀟依;龔駿毅 申請（專利權(quán)）人 廣西慧云信息技術(shù)有限公司
代理機(jī)構(gòu) 南寧東之智專利代理有限公司 代理人 戴燕桃;汪治興
地址 530007廣西壯族自治區(qū)南寧市高新區(qū)創(chuàng)新路23號9號樓三層
法律狀態(tài) -

摘要

摘要 本發(fā)明公開一種基于光照變換網(wǎng)絡(luò)的圖像自動增強(qiáng)方法，包括以下步驟：（1）數(shù)據(jù)集制作：收集大量圖像分類數(shù)據(jù)，并將圖像分類數(shù)據(jù)按0.8:0.1:0.1的比例隨機(jī)劃分為訓(xùn)練集、驗證集與測試集；（2）構(gòu)建一種光照變換網(wǎng)絡(luò)：所述光照變換網(wǎng)絡(luò)由圖像自動增強(qiáng)模塊與ResNet50模塊組成；所述光照變換網(wǎng)絡(luò)用于對輸入的原始圖像進(jìn)行自動增強(qiáng)；（3）模型訓(xùn)練：指定步驟(1)中的訓(xùn)練集與驗證集，使用SGD優(yōu)化器，將訓(xùn)練集輸入至步驟(2)的光照變換網(wǎng)絡(luò)，進(jìn)行模型訓(xùn)練；（4）模型推理。本發(fā)明根據(jù)輸入圖像進(jìn)行自動增強(qiáng)，可有效提升深度學(xué)習(xí)模型的光照一致性。??