├── README.md ├── caption-attention.py ├── caption-traditional.py ├── create_input_files.py ├── data └── WORDMAP_coco_5_cap_per_img_5_min_word_freq.json ├── datasets.py ├── download.sh ├── eval.py ├── modelAttention.py ├── modelTra.py ├── readme.txt ├── requirements.txt ├── train-attention.py ├── train-traditional.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/README.md -------------------------------------------------------------------------------- /caption-attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/caption-attention.py -------------------------------------------------------------------------------- /caption-traditional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/caption-traditional.py -------------------------------------------------------------------------------- /create_input_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/create_input_files.py -------------------------------------------------------------------------------- /data/WORDMAP_coco_5_cap_per_img_5_min_word_freq.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/data/WORDMAP_coco_5_cap_per_img_5_min_word_freq.json -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/datasets.py -------------------------------------------------------------------------------- /download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/download.sh -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/eval.py -------------------------------------------------------------------------------- /modelAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/modelAttention.py -------------------------------------------------------------------------------- /modelTra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/modelTra.py -------------------------------------------------------------------------------- /readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/readme.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | nltk 3 | numpy 4 | Pillow 5 | argparse 6 | tqdm 7 | -------------------------------------------------------------------------------- /train-attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/train-attention.py -------------------------------------------------------------------------------- /train-traditional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/train-traditional.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tqxg2018/Image-Caption-Pytorch/HEAD/utils.py --------------------------------------------------------------------------------