D2PRL

Requirements

Install the required packages by:

pip install -r requirements.txt

Pretrained models

Pretrain weight：https://pan.baidu.com/s/1m_FU5fqBl5J50uKs9Vu7Ng

code:xhwq

Test on Custom folder

Put forged images in testpic/ folder, and run

python test_demo.py

The output masks will be saved in output folder.

Train

If you want to train the model, please first download the training dataset and the pre-trained weights of SE-U-net, and then run train.py.

training dataset:https://pan.baidu.com/s/1PvSDcFiVmq5m_yFnp-RRVA

code:j49t

Pretrain weight of SE-U-net:https://pan.baidu.com/s/1tYIpEwxaPfZ64iCu06OTNg

code:i0i2