第57卷第1期2023年1月西安交通大学学报JOURNALOFXI’ANJIAOTONGUNIVERSITYVol.57No.1Jan.2023.*采用对比学习的多阶段Transformer图像去雾方法高峰1,2,汲胜昌1,郭洁1,侯杰3,欧阳超3,杨彪4(1.西安交通大学电气工程学院,710049,西安;2.国网陕西省电力有限公司电力科学研究院,710100,西安;3.哈尔滨工业大学(深圳)计算机科学与技术学院,518055,广东深圳;4.哈尔滨工业大学(深圳)建筑学院,518055,广东深圳)摘要:为了解决现有图像去雾方法在图像局部去雾以及纹理细节恢复等方面始终不理想以及处理非均匀雾质始终不彻底的问题,提出了一种采用对比学习的多阶段自注意力模块(Transformer)的图像去雾MSTCNet方法。首先,利用信道级Transformer模块作为基本的特征提取模块,充分地捕获特征信道之间的长距离依赖关系;其次,通过提出的多监督对比学习方法最大限度地挖掘正负样本信息,使去雾图像在投影后的隐空间中更靠近清晰图像,同时远离有雾图像;最后,利用多阶段渐进式网络结构和可变形自注意力机制有效地整合图像局部细粒度特征和全局粗粒度信息。本文在2个合成数据集和3个真实数据集上对所提出的方法进行了大量的实验,结果表明:所提出的MSTCNet方法在5个数据集上的峰值信噪比(PSNR)分别提高了1.49、1.45、0.11、1.45和0.22dB,在通用数据集与非数据集的测试中均超越已有的方法,在浓雾质、非均匀雾质以及均匀雾质的测试中均表现出最佳的去雾视觉效果,并达到最高的客观评价指标值。关键词:图像去雾;对比学习;自注意力;渐进式网络中图分类号:TP391文献标志码:ADOI:10.7652/xjtuxb202301019文章编号:0253-987X(2023)01-0195-16AMulti-StageTransformerNetworkforImageDehazingBasedonContrastiveLearningGAOFeng1,2,JIShengchang1,GUOJie1,HOUJie3,OUYANGChao3,YANGBiao4(1.SchoolofElectricalEngineering,Xi’anJiaotongUniversity,Xi’an710049,China;2.ElectricPowerResearchInstitute,StateGridShaanxiElectricPowerCompanyLimited,Xi’an710100,China;3.SchoolofComputerScience,HarbinInstituteofTechnology(Shenzhen),Shenzhen,Guangdong518055,China;4.SchoolofArchitecture,HarbinInstituteofTechnology(Shenzhen),Shenzhen,Guangdong518055,China)Abstract:Amulti-stageTransformernetworkforimagedehazingbasedoncontrastivelearningisproposedtosolvetheproblemthatexistingimagedehazingmethodsfailtoachievethedesiredre-sultsinlocalimagedehazingandde...
