SDRNet: An end-to-end shadow detection and removal network

https://doi.org/10.1016/j.image.2020.115832Get rights and content

Highlights

  • SDRNet can complete the shadow detection and removal tasks in a unified network.

  • SDRNet highly shares features and have no additional computational complexity.

  • A multi-scale global feature module is used to solve shadow area problems.

  • Shadow segmentation accuracy can be improved by using the image direction feature.

  • Superior performance can be achieved in both shadow detection and removal tasks.

Abstract

Image shadow detection and removal can effectively recover image information lost in the image due to the existence of shadows, which helps improve the accuracy of object detection, segmentation and tracking. Thus, aiming at the problem of the scale of the shadow in the image, and the inconsistency of the shadowed area with the original non-shadowed area after the shadow is removed, the multi-scale and global feature (MSGF) is used in the proposed method, combined with the non-local network and dense dilated convolution pyramid pooling network. Besides, aiming at the problem of inaccurate detection of weak shadows and complicated shape shadows in existing methods, the direction feature (DF) module is adopted to enhance the features of the shadow areas, thereby improving shadow segmentation accuracy. Based on the above two methods, an end-to-end shadow detection and removal network SDRNet is proposed. SDRNet completes the task of sharing two feature heights in a unified network without adding additional calculations. Experimental results on the two public datasets ISDT and SBU demonstrate that the proposed method achieves more than 10% improvement in the BER index for shadow detection and the RMSE index for shadow removal, which proves that the proposed SDRNet based on the MSGF module and DF module can achieve the best results compared with other existing methods.

Introduction

Image acquisition is the prerequisite for a variety of tasks. However, when acquiring images, external factors such as illumination and object occlusion results in shadows in the image. The existence of image shadows has advantages as well as disadvantages. The shadow can preserve the dynamic condition of the light scene and the valuable information of the objects in it. However, the shadow is a degrading phenomenon relative to the image, which can result in image information loss and influence the true information restoration of the original scene. It not only intuitively influences our observation for the image but also leads to subsequent image processing and analysis errors. Shadows are caused by an object occluding the light source, which is often combined with the corresponding occlusion object, has a very similar contour and is detrimental to the computer vision task. For instance, in the medical image processing field, shadows can interfere with lesion area segmentation and influence diagnosis. In the field of aerial image processing, shadows can influence ground target extraction, target matching, ground target tracking and many other tasks [1], [2]. For automatic driving, the existence of shadows can seriously influence the detection and segmentation of landmarks, stop lines, and lane lines [3]. To address the influence of image shadow and improve the practicality for a variety of applications, it is necessary to detect shadows in images, reconstruct the shadow area and remove the shadow, thereby reducing or eliminating the impact of the shadow on the computer vision task.

Shadow detection and removal consists of detection and removal of shadows in moving images (video sequences) and single images. The moving image can detect the shadow area by combining the relationship between the before-and-after-frame and then separating the shadow area from the object so that the target can be extracted accurately and conveniently. This method is mainly used for object tracking, such as vehicle tracking and human body tracking. Shadow removal for a single image is more complicated and more difficult because there is no before-and-after-frame information available. Thus, we mainly focus on this situation. Although there are many relevant studies worldwide, existing algorithms generally do not have universality; that is, they are only suitable for some specific scenarios. Therefore, studying a general single image shadow detection and removal algorithm that is suitable for a variety of scenes is still a challenging task. The contributions of this paper are mainly as follows.

(1) Considering the problem of the demand of the change in the shadow area for different receptive fields during the shadow detection process, and the inconsistency of the shadow area style and original non-shadowed area style in the de-shadowed images obtained by the existing de-shadowing algorithms, inspired by the non-local network and dense dilated convolutional pyramid pooling network, we propose a new module named the multi-scale and global feature (MSGF).

(2) Considering the problem of weak shadows and the complicated shape of shadows are difficult to accurately segment during the shadow detection process, we propose using the DF module to extract the image direction feature. By using the DF, the shadow area features can be enhanced, and the non-shadow area feature can be weakened. In addition, the interface between the shadow area and the non-shadow area can be strengthened, and the accuracy of the shadow segmentation can be greatly improved.

(3) We propose an end-to-end shadow detection and shadow removal network named SDRNet. SDRNet completes two tasks in a unified network, which highly share features and have no additional computational complexity. Experiments on public databases prove that SDRNet exceeds the existing methods in both shadow detection and shadow removal tasks, and optimum performance can be achieved by using SDRNet.

Section snippets

Related works

Most image de-shadowing studies first detect the shadow area in the image and then restore the shadow area. This kind of method can be divided into physical-based algorithms and deep learning-based algorithms.

SDRNet network structure

An end-to-end shadow detection and removal network SDRNet is proposed in this paper, which is different from the one-stage method and the two-stage method mentioned above. SDRNet integrates shadow detection and shadow removal into one network, which makes it possible for shadow detection and shadow removal to share features, greatly reducing the number of calculations and improving the efficiency of the model. At the same time, the joint training of the shadow detection network and shadow

Experimental results and analysis

In this section, we mainly conduct experiments on SDRNet and compare the shadow detection and shadow removal results with those of the existing optimal methods and separately perform quantitative and qualitative analyses.

Discussion

Image shadow detection and removal face many situations, apart from inaccurate shadow detection of weak shadows, complicated shadow shapes, and style inconsistency between shadow and non-shadow areas, overlapping shadows with different intensities is also a common phenomenon. In our experiment, we find that overlapping shadows with different intensities is in fact still a shadow. Possible situations including many weak shadows overlaps with each other may become a strong shadow. While the weak

Conclusion

As an important part of image pre-processing, image shadow detection and removal can effectively recover lost image information caused by the existence of shadows, which is helpful for improving the accuracy of object detection, segmentation and tracking. Therefore, we proposed an end-to-end shadow detection and removal network SDRNet to deal with the typical challenges for the two tasks, such as the inaccuracy of weak shadows, complicated shape shadow detection, and the style inconsistency

CRediT authorship contribution statement

Jin Tang: Supervision, Project administration. Qing Luo: Conceptualization, Methodology, Software, Writing - original draft. Fan Guo: Writing - original draft, Writing - review & editing. Zhihu Wu: Visualization, Investigation. Xiaoming Xiao: Data curation, Investigation. Yan Gao: Resources, Validation.

Acknowledgements

This work was supported by the National Natural Science Foundation of China under Grant 61502537, and the Natural Science Foundation of Hunan Province of China under Grant 2018JJ3681, 2016JJ2150.

References (31)

  • JiangC. et al.

    Shadow segmentation and classification in a constrained environment

    CVGIP: Image Underst.

    (1994)
  • SalvadorE. et al.

    Cast shadow segmentation using invariant color features

    Comput. Vis. Image Underst.

    (2004)
  • CucchiaraR. et al.

    Detecting moving objects, ghosts, and shadows in video streams

    IEEE Trans. Pattern Anal. Mach. Intell.

    (2003)
  • NadimiS. et al.

    Physical models for moving shadow and object detection in video

    IEEE Trans. Pattern Anal. Mach. Intell.

    (2004)
  • A. Yoneyama, C.H. Yeh, C.C.J. Kuo, Moving cast shadow elimination for robust vehicle extraction based on 2D joint...
  • E. Salvador, A. Cavallaro, T. Ebrahimi, Shadow identification and classification using invariant color models, in: 2001...
  • J. Zhu, K.G. Samuel, S.Z. Masood, et al. Learning to recognize shadows in monochromatic natural images, in: 2010 IEEE...
  • J.-F. Lalonde, A.A. Efros, S.G. Narasimhan, Detecting ground shadows in outdoor consumer photographs, in: European...
  • M. Baba, Asada N., Shadow removal from a real picture, in: Siggraph Conference on Sketches & Applications: in...
  • E. Arbel, Hel-Or H., Texture-preserving shadow removal in color images containing curved surfaces, in: 2007 IEEE...
  • F. Liu, M. Gleicher, Texture-consistent shadow removal, in: Computer Vision - ECCV 2008, Pt Iv, Proceedings, 2008, pp....
  • FinlaysonG.D. et al.

    Entropy minimization for shadow removal

    Int. J. Comput. Vis.

    (2009)
  • YangQ. et al.

    Shadow removal using bilateral filtering

    IEEE Trans. Image Process.

    (2012)
  • GuoR. et al.

    Paired regions for shadow detection and removal

    IEEE Trans. Pattern Anal. Mach. Intell.

    (2013)
  • H. Gong, D. Cosker, Interactive shadow editing from single images, in: C.V. Jawahar, S. Shan (Eds.), Computer Vision -...
  • Cited by (9)

    • Brightness-gradient difference feature guided shadow removal method

      2022, Knowledge-Based Systems
      Citation Excerpt :

      The first global network extracts shadow-related information and the second realizes shadow removal based on shadow-related information. In addition, Hu [37], Liu [38], Wang [39] and Tang [40] build Mask-ShadowGAN, Lightness-Guided Network, STackedCGAN and SDRNet to provide the results of shadow removal, respectively. After analyzing shadows in the image, Le [14] designs a shadow decomposition model, then uses a convolutional neural network to separate shadows from the outdoor image, and finally designs a shadow removal network to remove the separated shadows.

    • Hyperspectral image-based vegetation index (HSVI): A new vegetation index for urban ecological research

      2021, International Journal of Applied Earth Observation and Geoinformation
      Citation Excerpt :

      As cities expand, so does the number of high-rise towers and the shadowed area (Liu et al., 2011; Mostafa, 2017; Shahtahmassebi et al., 2013). However, since there is extremely little optical information in shadows, extracting shadowed vegetation is difficult (Sasi and Govindan, 2016; Tang et al., 2020; Wu et al., 2021). The emergence of new materials in cities, such as artificial sports surfaces in stadiums and blue roofs, also presents significant obstacles to vegetation extraction techniques (Guo and Li, 2020; Heiden et al., 2007; Herold et al., 2004).

    View all citing articles on Scopus

    No author associated with this paper has disclosed any potential or pertinent conflicts which may be perceived to have impending conflict with this work. For full disclosure statements refer to https://doi.org/10.1016/j.image.2020.115832.

    View full text