RICGA
Region Image Caption Generator
Annocement
This repo uses one of the official tensorflow models - im2txt and modifies its structure.
Usage
Download and preprocess MSCOCO data to ./ricga/data/mscoco/
using ./ricga/data/download_and_preprocess_mscoco.sh
Train the model using ./ricga/trian.py
Evaluate the mode in the same time using ./ricga/evaluate.py
View the tensorboard using ./ricga/tensroboard.sh
Run an inference sample using ./ricga/run_inference.py
Create model result in mscoco format for eval scores using ./ricga//create_results.py
Calculate model eval scores using ./ricga/eval_tools/cocoEvalCapDemo.ipynb
Run a inference server using ./ricga/ricga_server.py
original branch
The original branch is official tensorflow models - im2txt.