Home

Awesome

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.

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.

cd script
bash test.sh gpu_id

Prereuisites