- Python >= 3.6
- PyTorch >= 0.4.1
- OpenCV (Python version)
pip install -r requirements.txt
python test.py --root-path=$test set fold path --test-list=data/val_list.txt --checkpoint=models/baseline.pth.tar
cd evaluation
python evaluation.py --pre_rst_path=val_baseline.txt
This code is distributed under MIT LICENSE.
Please contact us if you have any problems during the demo running:
[email protected]