penguinbing / Mask-ShadowNet

PyTorch implementation of "Mask-ShadowNet: Towards Shadow Removal via Masked Adaptive Instance Normalization".

Mask-ShadowNet: Toward Shadow Removal via Masked Adaptive Instance Normalization

This is the Pytorch implementation of "Mask-ShadowNet: Toward Shadow Removal via Masked Adaptive Instance Normalization".

Note: The resolution of test shadow image should be 256*256 for our pre-trained model.

Please first set the dataset path then start training or testing.

• Training

cd script
bash train.sh gpu_id display_port

where gpu_id is the ID of gpu, and display_port is the port number of visdom server.

• Testing

cd script
bash test.sh gpu_id

Prereuisites

• Linux or macOS
• python 3
• CPU or NVIDIA GPU + CUDA CuDNN
• PyTorch 1.2+