Closed haoransh closed 5 years ago
By the way, I give a simple tutorial on how to use your pretrained model to perform table detection here https://github.com/haoransh/TableBank/blob/master/TestPretrainedModel.md. The detection on the sampled TableBank data looks perfect! Thanks for your great work!! Would you like to merge the repo maybe?
I read the document you wrote and it is pretty good. I'm not sure how to merge your repo, maybe you should start a new pull request?
Yeah, Thanks! I have created a pull request.
Merged.
python tools/infer_simple.py --cfg $MODEL_PATH --output-dir /tmp/detectron-tablebank --image-ext jpg \
--wts $CONFIG_MODEL /home/shr/TableBank/data/Sampled_Detection_data/Latex/images
maybe the upper line should change to:
python tools/infer_simple.py --cfg $CONFIG_MODEL --output-dir /tmp/detectron-tablebank --image-ext jpg \
--wts $MODEL_PATH /home/shr/TableBank/data/Sampled_Detection_data/Latex/images
Thanks! You are right. To be clear, --cfg
should be the YAML file of the model architecture, and --wts
should be the pickled file .pkl
of the model parameters.
I have fixed this error.
As topic. This can help to reproduce your experiments.