We believe our results can be reproduced, but if there is any problem please contact me
Download the test results from Google Drive.
Download the pretrained model from Google Drive, and place it in the folder saved_models.
Run the following code to generate test results.
Download the test dataset from Google Drive, and place it in the folder test_data.
Run the following code to generate test results.
BKchallenge
│ README.md
│
└───saved_models
│ │ boken_net_g_29500.pth
│ │ deboken_net_g_2000.pth
│ │ DIS_model.pth
│ │ model.pt
│
│
└───test_data
│ 00000.src.jpg
│ 00001.src.jpg
│ 00002.src.jpg
| ....
cd BKchallenge
sh test.sh
the results can be seen in the folder result.
you can look for the 'requirements.txt' to see the requirements
for a new environment you can run below codes
conda create --name boken_test python==3.8.15
conda activate boken_test
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch
pip install requirements.txt