xden2331 / img2caption

Image2Caption

Image2Caption is a web application for captioning one image.

New Features!

• Upload one image and get the caption automatically!

Tech

Dillinger uses a number of open source projects to work properly:

• [PyTorch] - The Deep Learning Lib.
• [Django] - Awesome web framework

Installation

Clone repository

$ git clone https://github.com/xden2331/img2caption.git

Go to the work directory

$ cd img2caption

Install virtual environment

$ pipenv install

Download the caption model

$ chmod a+x main/download_caption_model.sh
$ ./main/download_caption_model.sh

Install the dependencies and devDependencies and start the server.

$ pipenv install -r requirements.txt

Start Django Server

$ python manage.py runserver

Todos

• Integrate more image captioning models
• Deploy the web app

Acknowledge

MexsonFernandes - The whole project is modified on MexsonFernandes's project. David Chung - The idea of this project is given by David Chung.