模型与数据集链接: 提取码: jnp4
仪品网盘
解压后放在文件夹中即可
python app.py
- Linux
- CPU or NVIDIA GPU + CUDA CuDNN
- Python 3
- torch>=0.4.1
- torchvision>=0.2.1
- dominate>=2.3.1
- visdom>=0.1.8.3
- Train the model
bash ./train.sh
- Test
bash ./test.sh
- Evaluate
bash ./evaluate.sh
- scripts.sh integrate train.sh, test.sh, and evaluate.sh
bash ./scripts.sh