neural_joking_machine

This is a NN model for generating a Japanese funny caption for a image referenced on Neural Joking Machine : Humorous image captioning.
A lot of codes are from this repository.

Installation

git clone https://github.com/meciwo/neural_joking_machine.git
cd neural_joking_machine

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