TeamHiddenLeaf / Caption-Bot

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Caption-Bot

captionbot

Insipired by the Caption Bot implemented by Microsoft AI, we have made an attempt to replicate the application using deep learning models. Our project uses both methods from computer vision to understand the content of the image and a language model from the field of natural language processing to turn the understanding of the image into words in the right order.

Libraries/Parameters used in our project includes: Keras, Tensorflow, Word-embedding model (Glove), Images datasets from UIUC department of Computer Science (6000 trained images, 1000 test images)

Current Work: We are now working on the web appplication framework using Flask and finally deploy our application in Heroku Cloud. Also, we developed the entire code in Kaggle and we are in the process of transferring it to the Google Colaboratory. Therefore, Please regularly check for updates in the repository.

Access to the dataset: https://www.flickr.com/photos/tags/uiuc/

Happy Coding!

About

License:MIT License


Languages

Language:Jupyter Notebook 100.0%