Please use this identifier to cite or link to this item:
Title: Data augmentation and shadow image classification for shadow detection
Authors: Li, G
Wen, L
Huang, Z
Xia, R
Pang, Y
Issue Date: 1-Dec-2021
Publisher: John Wiley & Sons Ltd on behalf of The Institution of Engineering and Technology
Citation: Li, G., Wen, L., Huang, Z., Xia, R. and Pang, Y. (2021) 'Data augmentation and shadow image classification for shadow detection', IET Image Process. 0 (in press), pp. 1 - 12. doi: 10.1049/ipr2.12377.
Abstract: Shadow detection is an important branch of computer vision. Recently, convolutional neural network (CNN)-based methods for shadow detection have achieved better performance than methods based on manually designed features. However, CNNs are extremely hungry for data and the training of CNN-based shadow detector requires time-consuming and expensive pixel-level annotations. To alleviate this problem in shadow detection, a method of data augmentation based on generative adversarial network (GAN), named ShadowGAN, has been proposed. Given a shadow mask and a shadow-free image, our ShadowGAN can generate shadow images with labels. To guide the training of ShadowGAN and get more realistic shadow images, (Formula presented.) loss is further implemented to impose a restriction between real shadow images and generated shadow images. The effectiveness of ShadowGAN is demonstrated by training existing shadow detectors on enlarged dataset. In addition, to better make use of shadow-free images in shadow detection, shadow image classification task is added for the shadow detectors. Experiments show that this task can guide the feature extraction network to learn more robust shadow features. At last, these two methods are combined and a better performance of shadow detection is achieved.
ISSN: 1751-9659
Appears in Collections:Dept of Electronic and Electrical Engineering Research Papers

Files in This Item:
File Description SizeFormat 
FullText.pdf3.08 MBAdobe PDFView/Open

This item is licensed under a Creative Commons License Creative Commons