王圣亚
南京审计大学
摘要(Abstract):
本文提出了一种基于跨尺度残差注意力的生成对抗阴影去除算法(CSRA-GAN),用于高质量的阴影去除与细节恢复。该方法结合Transform架构与多尺度特征融合策略,并引入通道与局部高斯注意力机制,以增强网络的特征表达与细粒度信息建模能力。为进一步优化重建效果,设计了多生成器协同框架,实现阴影去除、逆光修复和细节增强的联合建模。在ISTD+与SRD数据集上的实验结果表明,该方法在结构相似性(SSIM)和峰值信噪比(PSNR)等指标上均显著优于现有方法,生成图像在视觉质量和细节保真度方面表现突出,验证了所提方法在阴影去除及图像重建任务中的有效性与泛化能力。
关键词(KeyWords):
阴影去除;深度学习;动态范围扩展;生成对抗网络;图像增强
参考文献(References):
[1]王嘉,关夏威,付少波,等.基于先验信息的光照鲁棒性水下视觉定位方法[J].舰船科学技术,2024,46(10):98-101.
[2]陈佛计,朱枫,吴清潇,等.生成对抗网络及其在图像生成中的应用研究综述[J].计算机学报,2021,44(02):347-369.
[3]张宸嘉,朱磊,俞璐.卷积神经网络中的注意力机制综述[J].计算机工程与应用,2021,57(20):64-72.
[4]Wang P,Zhao X.Multi-scale adaptive light propagation for shadow reduction[C].In Proceedings of the IEEE/CVF International Conference on Image Processing,2020,5032-5039.
[5]肖雅妮,范馨月,陈文峰.多分支融合局部特征的行人重识别算法[J].计算机工程与应用,2021,57(18):213-219.
[6]J.C,T.C,B.X,et al.SE-ECGNet:Multi-scale SE-Net for Multi-lead ECG Data[J].Computing in Cardiology,2020.
[7]凌艳,陈莹.多尺度上下文信息增强的显著目标检测全卷积网络[J].计算机辅助设计与图形学学报,2019,31(11):2007-2016.
[8]Yurui Zhu,Jie Huang,Xueyang Fu,Feng Zhao,Qibin Sun, and Zheng-Jun Zha.Bijective Mapping Network for Shadow Removal.In CVPR,2022.3-8.
[9]Fu L,Zhou C,Guo Q,et al.Auto-exposure fusion for single image shadow removal[C].In Proc.IEEE Int’l Conf.Computer Vision and Pattern Recognition,2021,10571-10580.
[10]Guo,L.;Wang,C.;Yang,W.;Huang,S.;Wang,Y.; Pfister,H.;and Wen,B.2023b.Shadowdiffusion:When degradation prior meets diffusion model for shadow removal.In IEEE Conference on C.
[11]Zhang H,Xu Y, Zhu S, et al.STC-GAN:Shadow transfer and removal via a generative adversarial network[J].IEEE Transactions on Image Processing.New York,NY:IEEE,2022,31:3844-3857.
[12]Yurui Zhu,Zeyu Xiao,Yanchi Fang,Xueyang Fu, Zhiwei Xiong,and Zheng-Jun Zha.Efficient model-driven network for shadow removal.In AAAI,2022,6-7.
[13]Guo R,Dai Q, Hoiem D.Paired regions for shadow detection and removal[J].IEEE transactions on pattern analysis and machine intelligence,2012,35(12):2956-2967.
[14]Le H,Samaras D.Shadow removal via shadow image decomposition[C].In Proceedings of the IEEE International Conference on Computer Vision,2019,8578-8587.
[15]Kangfu Mei,Luis Figueroa,Zhe Lin,Zhihong Ding, Scott Cohen,Vishal M.Patel;Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision(WACV),2024, pp.4313-4.