/img2caption

Primary LanguageJavaScriptGNU General Public License v3.0GPL-3.0

Image2Caption

Build Status

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.