Very Slow Processing Time

InnerPeace-Wu / densecap-tensorflow

Re-implement CVPR2017 paper: "dense captioning with joint inference and visual context" and minor changes in Tensorflow. (mAP 8.296 after 500k iters of training)

MIT License

61 stars 28 forks source link

Very Slow Processing Time #8

Closed choochtech closed 6 years ago

choochtech commented 6 years ago

Hi,

It takes more than 1 minute to process 1 image.

I ran the bash scripts/dense_cap_demo.sh ./output/ckpt ./output/ckpt/vocabulary.txt command.

I wonder if I am doing anything wrong. I am using an aws p2.xlarge gpu. Is this normal ?

Or do have to do another configuration I am not aware of. Please any help is appreciated.

Thanks

InnerPeace-Wu commented 6 years ago

Hi, I tested once, K80 is slower than gtx 10xx. You may change the config TEST.USE_BEAM_SEARCH in scripts/dense_cap_demo.sh to be False and it will be faster. best,

choochtech commented 6 years ago

Hi,

Thanks I did change it and it was a lot faster. Good job on the code ! Thanks

InnerPeace-Wu commented 6 years ago

Glad to help. I will close the issue, but feel free to reopen it if you have further question. best