- Uses pretrained ResNet50 model and Glove embeddings to caption any image
Model ARCHITECTURE img feature --------> MODEL --> Next word in sequence ---- partial sequence ---> | | _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ | Partial caption ----> RNN \ \ Feed forward network ----> predicted word,next / ending with softmax in the sequence of / partial caption Image vector
Link for this model
pip install tensorflow
pip install keras