Object Shadow Generation is to deal with the shadow inconsistency between the foreground object and the background in a composite image, that is, generating plausible shadow for the foreground object according to background information, to make the composite image more realistic.
Our dataset DESOBA is a synthesized dataset for Object Shadow Generation. We build our dataset on the basis of Shadow-OBject Association dataset SOBA, which collects real-world images in complex scenes and provides annotated masks for object-shadow pairs. Based on SOBA dataset, we remove all the shadows to construct our DEshadowed Shadow-OBject Association (DESOBA) dataset, which can be used for shadow generation task and other shadow-related tasks as well.
Illustration of DESOBA dataset construction (top flow): Given a ground-truth target image Ig, we manually remove all shadows to produce a deshadowed image Id. Then, we randomly select a foreground object in Ig, and replace its shadow area with the counterpart in Id to synthesize a composite image Ic without foreground shadow. Ic and Ig form a pair of input composite image and ground-truth target image.
Illustration of shadow generation task (bottom flow): Given Ic and its foreground object mask Mfo, we aim to reconstruct the target image Ig with foreground shadow.
Our DESOBA dataset contains 839 training images with totally 2,995 object-shadow pairs and 160 test images with totally 624 object-shadow pairs. Note that we discard one complex training image from SOBA. During testing, we ignore 9 object-shadow pairs with too small shadows after the whole image is resized to 256ร256, resulting in 615 test image pairs.The DESOBA dataset is provided in Baidu Cloud (access code: 1234), or Google Drive.
We refer to the test image pairs with Background Object-Shadow (BOS) pairs as BOS test image pairs (from left to right in the left subfigure: synthetic composite image, foreground object mask, background object mask,background shadow mask, and ground-truth target image), and the remaining ones as BOS-free test image pairs (from left to right in the right subfigure: synthetic composite image, foreground object mask, and ground-truth target image).
cd ./DESOBA_DATASET_util
- Visualizing training image pairs (839 train images with 11509 pairs):
python Vis_Desoba_Dataset --serial_batches --isTrain 1
Training image pairs are stored in /DESOBA_DATASET/TrainTestVisualization/train/.
- Visualizing BOS-free test image pairs (34 BOS-free test images with 34 pairs):
python Vis_Desoba_Dataset --serial_batches --isTrain 0 --bosfree
BOS-free test image pairs are stored in /DESOBA_DATASET/TrainTestVisualization/train/test_bosfree.
- Visualizing BOS test image pairs (126 BOS test images with 581 pairs):
python Vis_Desoba_Dataset --serial_batches --isTrain 0 --bos
BOS test image pairs are stored in /DESOBA_DATASET/TrainTestVisualization/train/test_bos.
We show some examples of training/testing tuples below:
From left to right: synthetic composite image without foreground shadow, target image with foreground shadow, foreground object mask, foreground shadow mask, background object mask, and background shadow mask.
cd ./DESOBA_DATASET_util
- Producing real composite images with one foreground object:
python Vis_RealCompositionImages_fromDesoba_Dataset.py --foreground_object_num 1
Real composite images with one foreground object are stored in /DESOBA_DATASET/CompositeImages/1_ForegroundObject/.
- Producing real composite images with two foreground objects:
python Vis_RealCompositionImages_fromDesoba_Dataset.py --foreground_object_num 2
Real composite images with two foreground objects are stored in /DESOBA_DATASET/CompositeImages/2_ForegroundObject.
We show a real composite image with one foreground object and a real composite image with two foreground objects below:
From left to right: synthetic composite image without foreground shadow, foreground object mask, background object mask, background shadow mask.
To evaluate the effectiveness of different methods in real scenarios, we prepare 100 real composite images including 74 images with one foreground object and 26 images with two foreground objects. We provide 100 real composite images in Baidu Cloud (access code: 1234), or Google Drive.
We provide the code of obtaining training/testing tuples, in which each tuple contains foreground object mask, foreground shadow mask, background object mask, background shadow mask, shadow image, and synthetic composite image without foreground shadow mask. The dataloader is available in /DESOBA_DATASET_util/data/DesobaSyntheticImageGeneration_dataset.py
, which can be used as dataloader in training phase or testing phase.
-
Pix2Pix: Image to image translation method. Implementation of paper "Image-to-Image Translation with Conditional Adversarial Nets" [pdf].
-
Pix2Pix-Res: Image to image translation method. Implementation of paper "Image-to-Image Translation with Conditional Adversarial Nets" [pdf]. Pix2Pix-Res is a variant of Pix2Pix whose architecture is the same as Pix2Pix but outputs the residual results.
-
ShadowGAN: Image to image translation method. Implementation of paper "ShadowGAN: Shadow synthesis for virtual objects with conditional adversarial networks" [pdf].
-
Mask-ShadowGAN: Image to image translation method. Implementation of paper "Mask-ShadowGAN: Learning to remove shadows from unpaired data" [pdf].
-
ARShadowGAN: Image to image translation method. Implementation of paper "ARShadowGAN: Shadow Generative Adversarial Network for Augmented Reality in Single Light Scenes" [pdf].
Here we provide PyTorch implementation and the trained model of our SGRNet.
- Python
- Pytorch
- PIL
- you can create a new environment by using
pip install -r requirements.txt
- Clone this repo:
git clone https://github.com/bcmi/Object-Shadow-Generation-Dataset-DESOBA.git
cd Object-Shadow-Generation-Dataset-DESOBA
-
Download the DESOBA dataset from Baidu Cloud (access code: 1234), or Google Drive. Save it at
/DESOBA_DATASET/
. -
Download the pretrained model of SGRNet and baseline methods from Baidu Cloud (access code:1234), or Google Drive. Save it at
/TrainedModels/
.
DESOBA dataset has six subfolders including: ShadowImage, DeshadowedImage, InstanceMask, ShadowMask, newshadowmask, shadowparams.
You should provide foreground object mask and background image without foreground shadow.
-
For SGRNet method:
-
With background object mask and background shadow mask, you can modify (reseting the paths of your own data) and run
/src/script/SGRNet_RealComposite.sh
to produce foreground shadow for your own composite images using pretrained SGRNet model.
- Without background object mask and background shadow mask, you can modify (reseting the paths of your own data) and run
/src/script/SGRNet_RealComposite_2.sh
to produce foreground shadow for your own composite images using pretrained SGRNet model.
- For baselines methods: You can run
/src/script/Pix2pix_RealComposite.sh
/src/script/Pix2pixRes_RealComposite.sh
/src/script/ARShadowGAN_RealComposite.sh
/src/script/MaskshadowGAN_RealComposite.sh
/src/script/ShadowGAN_RealComposite.sh
to produce foreground shadow for your own composite images using pretrained baseline models, respectively.
We provide 100 real composite images with foreground mask and you can download them to /DESOBA_DATASET/.
- For SGRNet method: You can run
/src/script/SGRNet_RealComposite.sh
to produce shadow for prepared real composite images using pretrained SGRNet model.
- For baselines methods: You can run,
/src/script/Pix2pix_RealComposite.sh
/src/script/Pix2pixRes_RealComposite.sh
/src/script/ARShadowGAN_RealComposite.sh
/src/script/MaskshadowGAN_RealComposite.sh
/src/script/ShadowGAN_RealComposite.sh
to produce shadow for prepared real composite images using pretrained baseline models.
-
For SGRNet method:
-
For BOS test images, set
TESTDATA=--bos
in/src/script/SGRNet_test.sh
and run
/src/script/SGRNet_test.sh
to conduct evaluation on BOS test images from DESOBA dataset.
- For BOS-free test images, set
TESTDATA=--bosfree
in/src/script/SGRNet_test.sh
and run
/src/script/SGRNet_test.sh
to conduct evaluation on BOS-free test images from DESOBA dataset.
-
For baselines methods:
-
For BOS test images, set
TESTDATA=--bos
and run
/src/script/Pix2pix_test.sh
/src/script/Pix2pixRes_test.sh
/src/script/ARShadowGAN_test.sh
/src/script/MaskshadowGAN_test.sh
/src/script/ShadowGAN_test.sh
to conduct evaluation on BOS test images from DESOBA dataset for different baseline methods, respectively.
- For BOS-free test images, set
TESTDATA=--bosfree
and run
/src/script/Pix2pix_test.sh
/src/script/Pix2pixRes_test.sh
/src/script/ARShadowGAN_test.sh
/src/script/MaskshadowGAN_test.sh
/src/script/ShadowGAN_test.sh
to conduct evaluation on BOS-free test images from DESOBA dataset for different baseline methods, respectively.
Considering that our SGRNet relied on shadow parameters during training phase, we need to calculate shadow parameters from paired shadow-deshadow images. We provide calculated shadow parameters in DESOBA dataset. You can also calculate shadow parameters based on tuple {shadowimage, deshadowed image, shadow mask} by runing /data_processing/compute_params_DESOBA.py
- For SGRNet method: you can run
/src/script/SGRNet_train.sh
to train SGRNet model on DESOBA dataset.
- For baselines methods: you can run
/src/script/Pix2pix_train.sh
/src/script/Pix2pixRes_train.sh
/src/script/ARShadowGAN_train.sh
/src/script/MaskshadowGAN_train.sh
/src/script/ShadowGAN_train.sh
to train model on DESOBA dataset for different baseline methods, respectively.
Here we show the quantitative results of different methods on DESOBA test set based on Root Mean Square Error (RMSE) and Structural SIMilarity index (SSIM). Global RMSE (GRMSE) and Global SSIM (GSSIM) are calculated over the whole image, while Local RMSE (LRMSE) and Local SSIM (LSSIM) are calculated over the ground-truth foreground shadow area. The paper/code/model of shadow generation related methods are summarized in Awesome-Object-Shadow-Generation.
Method | BOS Test Images | BOS-free Test Images | ||||||
---|---|---|---|---|---|---|---|---|
Evaluation metric | GRMSE | LRMSE | GSSIM | LSSIM* | GRMSE | LRMSE | GSSIM | LSSIM* |
Pix2Pix | 7.659 | 75.346 | 0.927 | 0.249 | 18.875 | 81.444 | 0.856 | 0.110 |
Pix2Pix-Res | 5.961 | 76.046 | 0.971 | 0.253 | 18.305 | 81.966 | 0.901 | 0.107 |
ShadowGAN | 5.985 | 78.413 | 0.986 | 0.240 | 19.306 | 87.017 | 0.918 | 0.078 |
Mask-ShadowGAN | 8.287 | 79.212 | 0.953 | 0.245 | 19.475 | 83.457 | 0.891 | 0.109 |
ARShandowGAN | 6.481 | 75.099 | 0.983 | 0.251 | 18.723 | 81.272 | 0.917 | 0.109 |
SGRNet | 4.754 | 61.762 | 0.988 | 0.380 | 15.128 | 61.439 | 0.927 | 0.183 |
*: Note that the LSSIM results in official AAAI paper are miscalculated due to a bug in the evaluation code util/ssim.py. We sincerely apologize for this mistake and have updated the results in arXiv.
Here we show some example results of different baselines on DESOBA dataset. More examples can be found in our main paper.
From left to right: input composite image (a), foreground object mask (b), results of Pix2Pix (c), Pix2Pix-Res (d), ShadowGAN (e), Mask-ShadowGAN (f), ARShadowGAN (g), our SGRNet (h), ground-truth (i). The results on BOS test images are shown in row 1-2, and the results on BOS-free test images are shown in row 3-4).
Below we present several results of different baselines on real composite images. The 100 real composite images could be found in Baidu Cloud (access code: 1234), or Google Drive.
From left to right: input composite image (a), foreground object mask (b), results of Pix2Pix (c), Pix2Pix-Res (d), ShadowGAN (e), Mask-ShadowGAN (f), ARShadowGAN (g), SGRNet (h).
If you find this work is useful for your research, please cite our paper using the following BibTeX [arxiv]:
@article{hong2021shadow,
title={Shadow Generation for Composite Image in Real-world Scenes},
author={Hong, Yan and Niu, Li and Zhang, Jianfu},
journal={AAAI},
year={2022}
}