The code for deploying the NIC framework on AoANet.
- Python 3.6
- Java 1.8.0
- PyTorch 1.7.0
- cider (already been added as a submodule)
- coco-caption (already been added as a submodule)
- tensorboardX
You should preprocess the dataset and get the cache for calculating cider score for SCST:
$ python scripts/prepro_ngrams.py --input_json data/dataset_coco.json --dict_json data/cocotalk.json --output_pkl data/coco-train --split train
$ sh train.sh
This repository is based on AoANet , and you may refer to it for more details about the code.