├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── data.py ├── demo ├── demo-ids.txt └── demo-predictions.json ├── docs ├── baseline.jpg ├── features.jpg └── predictions.png ├── environment.yml ├── generate.py ├── model ├── caption.py ├── inception.py └── modules.py ├── preprocess ├── preprocess_captions.sh ├── preprocess_features.py ├── preprocess_images.py ├── tokenize_captions.py └── tokenize_captions_scst.py ├── preprocess_captions.sh ├── preprocess_features.sh ├── preprocess_images.sh ├── score.py ├── score.sh ├── scst ├── criterion.py └── generator.py ├── splits ├── karpathy_test_images.txt ├── karpathy_train_images.txt └── karpathy_valid_images.txt ├── task.py └── viewer.ipynb /.gitignore: -------------------------------------------------------------------------------- 1 | .idea -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/README.md -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/data.py -------------------------------------------------------------------------------- /demo/demo-ids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/demo/demo-ids.txt -------------------------------------------------------------------------------- /demo/demo-predictions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/demo/demo-predictions.json -------------------------------------------------------------------------------- /docs/baseline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/docs/baseline.jpg -------------------------------------------------------------------------------- /docs/features.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/docs/features.jpg -------------------------------------------------------------------------------- /docs/predictions.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/docs/predictions.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/environment.yml -------------------------------------------------------------------------------- /generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/generate.py -------------------------------------------------------------------------------- /model/caption.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/model/caption.py -------------------------------------------------------------------------------- /model/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/model/inception.py -------------------------------------------------------------------------------- /model/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/model/modules.py -------------------------------------------------------------------------------- /preprocess/preprocess_captions.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/preprocess/preprocess_captions.sh -------------------------------------------------------------------------------- /preprocess/preprocess_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/preprocess/preprocess_features.py -------------------------------------------------------------------------------- /preprocess/preprocess_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/preprocess/preprocess_images.py -------------------------------------------------------------------------------- /preprocess/tokenize_captions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/preprocess/tokenize_captions.py -------------------------------------------------------------------------------- /preprocess/tokenize_captions_scst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/preprocess/tokenize_captions_scst.py -------------------------------------------------------------------------------- /preprocess_captions.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/preprocess_captions.sh -------------------------------------------------------------------------------- /preprocess_features.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/preprocess_features.sh -------------------------------------------------------------------------------- /preprocess_images.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/preprocess_images.sh -------------------------------------------------------------------------------- /score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/score.py -------------------------------------------------------------------------------- /score.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash -e 2 | 3 | export PYTHONPATH=./external/coco-caption 4 | 5 | python score.py "$@" 6 | -------------------------------------------------------------------------------- /scst/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/scst/criterion.py -------------------------------------------------------------------------------- /scst/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/scst/generator.py -------------------------------------------------------------------------------- /splits/karpathy_test_images.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/splits/karpathy_test_images.txt -------------------------------------------------------------------------------- /splits/karpathy_train_images.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/splits/karpathy_train_images.txt -------------------------------------------------------------------------------- /splits/karpathy_valid_images.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/splits/karpathy_valid_images.txt -------------------------------------------------------------------------------- /task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/task.py -------------------------------------------------------------------------------- /viewer.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/krasserm/fairseq-image-captioning/HEAD/viewer.ipynb --------------------------------------------------------------------------------