SwinShadow

Requirement

• Python 3.10
• Pytorch 2.0.1

pip install -r requirements.txt

Datasets

• SBU dataset code:gxqt
• ISTD, ISTD+ dataset code:gxqt
• UCF dataset code:gxqt

Training

download the pretrain Swin Transformer backbone from here (code:gxqt), put it in the model folder

1. training on sbu dataset

modify tool/dataset.py line 27 and line 28, run python sbu.py, the ckpt will be saved in ./sbu/ckpt folder

2. training on istd dataset

modify tool/istddata.py line 27 and line 28, run python istd.py, the ckpt will be saved in ./istd/ckpt folder

Testing

run python tistd.py/tsbu.py/tucf.py, the file will be saved in results/xxx

Results

You can download the SBU, ISTD, and UCF results of our method from this link (code:gxqt).

Evaluation

the evaluation code is in the evaluation folder, using matlab to evaluate the results.