search

kdexd / virtex

[CVPR 2021] VirTex: Learning Visual Representations from Textual Annotations
http://kdexd.xyz/virtex
MIT License
556 stars 61 forks source link

Running Image Captioning Inference on Arbitrary Images: FileNotFound #13

Closed xden2331 closed 3 years ago

xden2331 commented 4 years ago

python scripts/eval_captioning.py \ --config /tmp/bicaptioning_R_50_L1_H2048/pretrain_config.yaml \ --checkpoint-path /tmp/bicaptioning_R_50_L1_H2048/checkpoint_500000.pth \ --data-root /path/to/images_dir \ --output /path/to/save/predictions.json \ --num-gpus-per-machine 1 \ --cpu-workers 4

Hi, I am wondering where can I download the pretrain_config.yaml and the checkpoint file. I cannot find the link on Github README. Is the only way to get the pre-trained model retraining the whole model?

Cheers,

kdexd commented 4 years ago

Hi, you can download the model either as shown by the code example in Model Zoo, or by download URLs in Model Zoo

The corresponding config files are present in configs directory. Please check our Model Zoo docs and match the config file names with the download URLs.

Let me know if you have further questions, thanks!

kdexd commented 3 years ago

I am assuming that it worked out for you; I am closing this issue due to lack of activity for a month. Feel free to reopen if you have any concerns, thanks!