poojahira / image-captioning-bottom-up-top-down

PyTorch implementation of Image captioning with Bottom-up, Top-down Attention
164 stars 39 forks source link

Captioning a new image #17

Closed piperino11 closed 4 years ago

piperino11 commented 4 years ago

If I want generate a caption for my image, where I can find the model fo faster R-ccn for extract the features?

ada-dl commented 4 years ago

I am new to deep learning and I would really want to know this as well. How can we run this model, once trained, on a new image, or train on a new dataset? Thank you anyone who can provide some insight on this.

poojahira commented 4 years ago

Hi, for a new image, you will need to extract the bottom up features from a Faster RCNN model first before using this codebase.