├── .gitignore ├── LICENSE ├── README.md ├── analyze_data.py ├── config.py ├── data_generator.py ├── demo.py ├── eval.py ├── font ├── WenQuanYiMicroHei-01.ttf ├── WenQuanYiMicroHeiMono-02.ttf └── simhei.ttf ├── images ├── dataset.png ├── image_0.jpg ├── image_1.jpg ├── image_2.jpg ├── image_3.jpg ├── image_4.jpg ├── image_5.jpg ├── image_6.jpg ├── image_7.jpg ├── image_8.jpg ├── image_9.jpg ├── net.png ├── out_0.jpg ├── out_1.jpg ├── out_2.jpg ├── out_3.jpg ├── out_4.jpg ├── out_5.jpg ├── out_6.jpg ├── out_7.jpg ├── out_8.jpg └── out_9.jpg ├── models.py ├── pre_process.py ├── requirements.txt ├── sponsor.jpg ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea 2 | __pycache__/ 3 | logs/ 4 | models/ 5 | data/ -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /analyze_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/analyze_data.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/config.py -------------------------------------------------------------------------------- /data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/data_generator.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/demo.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/eval.py -------------------------------------------------------------------------------- /font/WenQuanYiMicroHei-01.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/font/WenQuanYiMicroHei-01.ttf -------------------------------------------------------------------------------- /font/WenQuanYiMicroHeiMono-02.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/font/WenQuanYiMicroHeiMono-02.ttf -------------------------------------------------------------------------------- /font/simhei.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/font/simhei.ttf -------------------------------------------------------------------------------- /images/dataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/dataset.png -------------------------------------------------------------------------------- /images/image_0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_0.jpg -------------------------------------------------------------------------------- /images/image_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_1.jpg -------------------------------------------------------------------------------- /images/image_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_2.jpg -------------------------------------------------------------------------------- /images/image_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_3.jpg -------------------------------------------------------------------------------- /images/image_4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_4.jpg -------------------------------------------------------------------------------- /images/image_5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_5.jpg -------------------------------------------------------------------------------- /images/image_6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_6.jpg -------------------------------------------------------------------------------- /images/image_7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_7.jpg -------------------------------------------------------------------------------- /images/image_8.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_8.jpg -------------------------------------------------------------------------------- /images/image_9.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/image_9.jpg -------------------------------------------------------------------------------- /images/net.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/net.png -------------------------------------------------------------------------------- /images/out_0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_0.jpg -------------------------------------------------------------------------------- /images/out_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_1.jpg -------------------------------------------------------------------------------- /images/out_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_2.jpg -------------------------------------------------------------------------------- /images/out_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_3.jpg -------------------------------------------------------------------------------- /images/out_4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_4.jpg -------------------------------------------------------------------------------- /images/out_5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_5.jpg -------------------------------------------------------------------------------- /images/out_6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_6.jpg -------------------------------------------------------------------------------- /images/out_7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_7.jpg -------------------------------------------------------------------------------- /images/out_8.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_8.jpg -------------------------------------------------------------------------------- /images/out_9.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/images/out_9.jpg -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/models.py -------------------------------------------------------------------------------- /pre_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/pre_process.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | jieba 2 | gensim 3 | tqdm -------------------------------------------------------------------------------- /sponsor.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/sponsor.jpg -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/foamliu/Image-Captioning-PyTorch/HEAD/utils.py --------------------------------------------------------------------------------