Computer Science ›› 2021, Vol. 48 ›› Issue (6): 145-152.doi: 10.11896/jsjkx.200900109

• Computer Graphics & Multimedia • Previous Articles     Next Articles

Image Shadow Removal Algorithm Based on Generative Adversarial Network

SHI Heng, ZHANG Ling   

  1. School of Computer Science and Technology,Wuhan University of Science and Technology,Wuhan 430065,China
  • Received:2020-09-14 Revised:2021-03-10 Online:2021-06-15 Published:2021-06-03
  • About author:SHI Heng,born in 1996,postgraduate.His main reserch interests include image processing and computer gra-phics.(shihengh@163.com)
    ZHANG Ling,born in 1986,Ph.D.Her main research interests include image and video processing and computer graphics.
  • Supported by:
    National Natural Science Foundation of China(61902286).

Abstract: Although the existing learning-based shadow removal methods have made some progress,these methods mainly focus on the image itself and do not well explore other information related to shadow.These methods often result in problems such as image texture blurring or illumination inconsistency.To solve these problems,this paper proposes a new network based on genera-tive adversarial network (GAN) to remove shadows in the image.First,it uses an encoder-decoder to obtain a coarse shadow removal result.Then,it optimizes the coarse result by utilizing the shadow-related residual information to produce more realistic and natural shadow removal image.Our generator contains three encoder-decoder structures.The first encoder-decoder is used to restore the illumination of the image and generate a coarse shadow removal result.To solve the problem of color and illumination inconsistency,the residual information is input into the following encoder-decoder to correct the coarse results.Furthermore,the third encoder-decoder is used to refine the details in the image,which can avoid texture inconsistency between shadow regions and nonshadow regions.The discriminator is used to identify the authenticity of the image shadow removal result.Experiments show that the proposed method achieves the best RMSE value in both shaded region and non-shaded region,and effectively solves the problem of texture blur.In addition,the shadow removal image generated by the proposed method is much closer to the real image,which proves the effectiveness and feasibility of the proposed method.

Key words: Encoding-decoding structure, Generative Adversarial Network, Patch discrimi-nator, Residual information, Shadow removal

CLC Number: 

  • TP391
[1]KHAN S H,BENNAMOUN M,SOHEL F,et al.AutomaticShadow Detection and Removal from a Single Image[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2016,38(3):431-446.
[2]WEI J,LONG C,ZOU H,et al.Shadow Inpainting and Removal Using Generative Adversarial Networks with Slice Convolutions[J].Computer Graphics Forum,2019,38(7):381-392.
[3]IIZUKA S,SIMOSERRA E,ISHIKAWA H,et al.Globally and locally consistent image completion[J].ACM Transactions on Graphics,2017,36(4).
[4]WANG T,LIU M,ZHU J,et al.High-Resolution Image Synthesis and Semantic Manipulation with Conditional GANs[C]//Computer Vision and Pattern Recognition.2018:8798-8807.
[5]RAJ N B,VENKATESWARAN N.Single Image Haze Removal using a Generative Adversarial Network.[J].arXiv:1810.09479.
[6]ENGIN D,GENC A,EKENEL H K,et al.Cycle-Dehaze:En-hanced CycleGAN for Single Image Dehazing[C]//Computer Vision and Pattern Recognition.2018:825-833.
[7]LE H,VICENTE T F,NGUYEN V,et al.A+D Net:Training a Shadow Detector with Adversarial Shadow Attenuation[C]//European Conference on Computer Vision.2018:680-696.
[8]QIAN R,TAN R T,YANG W,et al.Attentive Generative Adversarial Network for Raindrop Removal from A Single Image[C]//Computer Vision and Pattern Recognition.2018:2482-2491.
[9]ISOLA P,ZHU J,ZHOU T,et al.Image-to-Image Translation with Conditional Adversarial Networks[C]//Computer Vision and Pattern Recognition,2017:5967-5976.
[10]LIU Y,LI Y S,ZHANG D P.Shadow elimination method based on chromaticity distortion and texture features[J].Computer Science,2005,32(9):211-214.
[11]CHEN R,LI P,HUANG Y,et al.Motion shadow removal algorithm based on multi-feature fusion[J].Computer Science,2018,45(6):291-295.
[12]GUO R,DAI Q,HOIEM D,et al.Single-image shadow detection and removal using paired regions[C]//Computer Vision and Pattern Recognition,2011:2033-2040.
[13]XIAO C,XIAO D,ZHANG L,et al.Efficient Shadow Removal Using Subregion Matching Illumination Transfer[J].Computer Graphics Forum,2013,32(7):421-430.
[14]FINLAYSON G D,HORDLEY S D,LU C,et al.On the remo-val of shadows from images[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2006,28(1):59-68.
[15]QU L,TIAN J,HE S,et al.DeshadowNet:A Multi-context Embedding Deep Network for Shadow Removal[C]//Computer Vision and Pattern Recognition.2017:2308-2316.
[16]HU X,FU C,ZHU L,et al.Direction-aware Spatial ContextFeatures for Shadow Detection and Removal[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2018:7454-7462.
[17]WANG J,LI X,YANG J,et al.Stacked Conditional Generative Adversarial Networks for Jointly Learning Shadow Detection and Shadow Removal[C]//Computer Vision and Pattern Recognition.2018:1788-1797.
[18]SIDOROV O.Conditional GANs for Multi-Illuminant ColorConstancy:Revolution or Yet Another Approach? [C]//The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) Workshops.IEEE,2019.
[19]DING B,LONG C,ZHANG L,et al.ARGAN:Attentive Recurrent Generative Adversarial Network for Shadow Detection and Removal[J].arXiv:1908.01323.
[20]ZHANG L,ZHANG Q,XIAO C,et al.Shadow Remover:Image Shadow Removal Based on Illumination Recovering Optimization[J].IEEE Transactions on Image Processing,2015,24(11):4623-4636.
[21]ZHANG L,LONG C,ZHANG X,et al.RIS-GAN:Explore Re-sidual and Illumination with Generative Adversarial Networks for Shadow Removal[C]//National Conference on Artificial Intelligence.2020.
[22]LIU F Q,LI Z H.Shadow removal method based on light-independent image[J].Journal of Image and Graphics,2007,12(10):1837-1840.
[23]XIAO M,HAN C Z.Edge-based motion shadow removal algorithm in indoor video[J].Pattern Recognition and Artificial Intelligence,2006,19(5):640-644.
[24]ZHANG H,PATEL V M.Density-Aware Single Image De-raining Using a Multi-stream Dense Network[C]//Computer Vision and Pattern Recognition.2018:695-704.
[25]GUO L S,GUO L,JIAO R H,et al.A target detection algorithm based on moving shadows[J].Pattern Recognition and Artificial Intelligence,2007,20(2):180-184.
[1] ZHANG Jia, DONG Shou-bin. Cross-domain Recommendation Based on Review Aspect-level User Preference Transfer [J]. Computer Science, 2022, 49(9): 41-47.
[2] SUN Qi, JI Gen-lin, ZHANG Jie. Non-local Attention Based Generative Adversarial Network for Video Abnormal Event Detection [J]. Computer Science, 2022, 49(8): 172-177.
[3] DAI Zhao-xia, LI Jin-xin, ZHANG Xiang-dong, XU Xu, MEI Lin, ZHANG Liang. Super-resolution Reconstruction of MRI Based on DNGAN [J]. Computer Science, 2022, 49(7): 113-119.
[4] XU Guo-ning, CHEN Yi-peng, CHEN Yi-ming, CHEN Jin-yin, WEN Hao. Data Debiasing Method Based on Constrained Optimized Generative Adversarial Networks [J]. Computer Science, 2022, 49(6A): 184-190.
[5] YIN Wen-bing, GAO Ge, ZENG Bang, WANG Xiao, CHEN Yi. Speech Enhancement Based on Time-Frequency Domain GAN [J]. Computer Science, 2022, 49(6): 187-192.
[6] XU Hui, KANG Jin-meng, ZHANG Jia-wan. Digital Mural Inpainting Method Based on Feature Perception [J]. Computer Science, 2022, 49(6): 217-223.
[7] DOU Zhi, WANG Ning, WANG Shi-jie, WANG Zhi-hui, LI Hao-jie. Sketch Colorization Method with Drawing Prior [J]. Computer Science, 2022, 49(4): 195-202.
[8] GAO Zhi-yu, WANG Tian-jing, WANG Yue, SHEN Hang, BAI Guang-wei. Traffic Prediction Method for 5G Network Based on Generative Adversarial Network [J]. Computer Science, 2022, 49(4): 321-328.
[9] LI Si-quan, WAN Yong-jing, JIANG Cui-ling. Multiple Fundamental Frequency Estimation Algorithm Based on Generative Adversarial Networks for Image Removal [J]. Computer Science, 2022, 49(3): 179-184.
[10] QU Zhong, CHEN Wen. Concrete Pavement Crack Detection Based on Dilated Convolution and Multi-features Fusion [J]. Computer Science, 2022, 49(3): 192-196.
[11] SHI Da, LU Tian-liang, DU Yan-hui, ZHANG Jian-ling, BAO Yu-xuan. Generation Model of Gender-forged Face Image Based on Improved CycleGAN [J]. Computer Science, 2022, 49(2): 31-39.
[12] LI Jian, GUO Yan-ming, YU Tian-yuan, WU Yu-lun, WANG Xiang-han, LAO Song-yang. Multi-target Category Adversarial Example Generating Algorithm Based on GAN [J]. Computer Science, 2022, 49(2): 83-91.
[13] TAN Xin-yue, HE Xiao-hai, WANG Zheng-yong, LUO Xiao-dong, QING Lin-bo. Text-to-Image Generation Technology Based on Transformer Cross Attention [J]. Computer Science, 2022, 49(2): 107-115.
[14] CHEN Gui-qiang, HE Jun. Study on Super-resolution Reconstruction Algorithm of Remote Sensing Images in Natural Scene [J]. Computer Science, 2022, 49(2): 116-122.
[15] JIANG Zong-li, FAN Ke, ZHANG Jin-li. Generative Adversarial Network and Meta-path Based Heterogeneous Network Representation Learning [J]. Computer Science, 2022, 49(1): 133-139.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!