Knowledge-Base Meme Caption Generator

This is a NN model for generating a Japanese funny caption for a image

Installation

git clone https://github.com/meciwo/Knowledge-based_Meme_Caption_Generator.git
cd KMCG

if you have already installed poetry

poetry install

if you havn't installed poetry

pip install poetry
poetry install

Usage

Scrayping

python scr/scrayping.py

preprocessing

python src/build_vocab.py

training

python src/train.py

sampling

python src/sampling --image PATH_TO_YOUR_IMAGE