AoANet:论文代码“注意图像字幕的注意”。 ICCV 2019-源码

注意图像字幕注意 该存储库包括“的实现。 要求 Python 3.6 Java 1.8.0 PyTorch 1.0 苹果酒（已经添加为子模块） coco-caption（已经添加为子模块） 张量板 培训AoANet 准备数据 请参阅data/README.md详细信息。 （注释：设置word_count_threshold在scripts/prepro_labels.py至4，以产生大小10369的词汇表。） 您还应该预处理数据集并获取缓存以计算苹果酒得分： $ python scripts/prepro_ngrams.py --input_json data/dataset_coco.json --dict_json data/cocotalk.json --output_pkl data/coco-train --split train 开始训练 $ CUDA_VISIB