基于空洞全局注意力機制的近紅外圖像彩色化方法

摘要： 針對目前卷積神經(jīng)網(wǎng)絡(luò)未能充分提取圖像的淺層特征信息導(dǎo)致近紅外圖像彩色化算法存在結(jié)果圖像局部區(qū)域誤著色及網(wǎng)絡(luò)訓(xùn)練不穩(wěn)定導(dǎo)致結(jié)果出現(xiàn)模糊問題，提出了一種新的生成對抗網(wǎng)絡(luò)方法用于彩色化任務(wù)。首先，在生成器殘差塊中引入自行設(shè)計的空洞全局注意力模塊，對近紅外圖像的每個位置理解更加充分，改善局部區(qū)域誤著色問題；其次，在判別網(wǎng)絡(luò)中，將批量歸一化層替換成梯度歸一化層，提升網(wǎng)絡(luò)判別性能，改善彩色...