search

wangyuxin87 / ContourNet

A PyTorch implementation of "ContourNet: Taking a Further Step toward Accurate Arbitrary-shaped Scene Text Detection" (CVPR2020)
Other
227 stars 46 forks source link

Test Error: ONLY ic15 model loads by default #33

Closed abdur75648 closed 3 years ago

abdur75648 commented 3 years ago

I want to test your trained model on CTW1500 data. So, I keep your model in output/ctw folder and run the command:

CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --nproc_per_node=1 tools/test_net.py --config-file "configs/ctw/r50_baseline.yaml"

But it loads the model _output/ic15/modelic15.pth by default (And gives a path error if I don't put model_ic15.pth there). It seems this has been hardcoded somewhere in the code but I'm NOT able to find it. Please tell me where is the correction required? @wangyuxin87