pip install torch
pip install torchvision
pip install tensorboardX
pip install tensorflow-cpu
pip install tqdm, imageio,
The dataset can be found here. The author just provide the download link, So I download the image for your convenience, the image data can be found in BaiduYunDisk
mkdir checkpoints
mkdir VisImage
mkdir -p runninglogs/runs
mkdir -p runinglogs/save
ln -s /yourdata ./data/processed/VP_Img_resize
- Training
- sh train.sh
- Test
- sh test.sh