├── LICENSE ├── README.md ├── data ├── __init__.py ├── dataset.py ├── example.py ├── field.py ├── utils.py └── vocab.py ├── environment.yml ├── evaluation ├── __init__.py ├── bleu │ ├── __init__.py │ ├── bleu.py │ └── bleu_scorer.py ├── cider │ ├── __init__.py │ ├── cider.py │ └── cider_scorer.py ├── meteor │ ├── __init__.py │ └── meteor.py ├── rouge │ ├── __init__.py │ └── rouge.py ├── stanford-corenlp-3.4.1.jar └── tokenizer.py ├── images ├── m2.png └── results.png ├── models ├── __init__.py ├── beam_search │ ├── __init__.py │ └── beam_search.py ├── captioning_model.py ├── containers.py └── transformer │ ├── __init__.py │ ├── attention.py │ ├── decoders.py │ ├── encoders.py │ ├── transformer.py │ └── utils.py ├── output_logs └── meshed_memory_transformer_test_o ├── test.py ├── train.py ├── utils ├── __init__.py ├── typing.py └── utils.py └── vocab.pkl /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/data/dataset.py -------------------------------------------------------------------------------- /data/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/data/example.py -------------------------------------------------------------------------------- /data/field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/data/field.py -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/data/utils.py -------------------------------------------------------------------------------- /data/vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/data/vocab.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/environment.yml -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/__init__.py -------------------------------------------------------------------------------- /evaluation/bleu/__init__.py: -------------------------------------------------------------------------------- 1 | from .bleu import Bleu -------------------------------------------------------------------------------- /evaluation/bleu/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/bleu/bleu.py -------------------------------------------------------------------------------- /evaluation/bleu/bleu_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/bleu/bleu_scorer.py -------------------------------------------------------------------------------- /evaluation/cider/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/cider/__init__.py -------------------------------------------------------------------------------- /evaluation/cider/cider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/cider/cider.py -------------------------------------------------------------------------------- /evaluation/cider/cider_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/cider/cider_scorer.py -------------------------------------------------------------------------------- /evaluation/meteor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/meteor/__init__.py -------------------------------------------------------------------------------- /evaluation/meteor/meteor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/meteor/meteor.py -------------------------------------------------------------------------------- /evaluation/rouge/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/rouge/__init__.py -------------------------------------------------------------------------------- /evaluation/rouge/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/rouge/rouge.py -------------------------------------------------------------------------------- /evaluation/stanford-corenlp-3.4.1.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/stanford-corenlp-3.4.1.jar -------------------------------------------------------------------------------- /evaluation/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/evaluation/tokenizer.py -------------------------------------------------------------------------------- /images/m2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/images/m2.png -------------------------------------------------------------------------------- /images/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/images/results.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/beam_search/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/beam_search/__init__.py -------------------------------------------------------------------------------- /models/beam_search/beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/beam_search/beam_search.py -------------------------------------------------------------------------------- /models/captioning_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/captioning_model.py -------------------------------------------------------------------------------- /models/containers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/containers.py -------------------------------------------------------------------------------- /models/transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/transformer/__init__.py -------------------------------------------------------------------------------- /models/transformer/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/transformer/attention.py -------------------------------------------------------------------------------- /models/transformer/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/transformer/decoders.py -------------------------------------------------------------------------------- /models/transformer/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/transformer/encoders.py -------------------------------------------------------------------------------- /models/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/transformer/transformer.py -------------------------------------------------------------------------------- /models/transformer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/models/transformer/utils.py -------------------------------------------------------------------------------- /output_logs/meshed_memory_transformer_test_o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/output_logs/meshed_memory_transformer_test_o -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/utils/typing.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/utils/utils.py -------------------------------------------------------------------------------- /vocab.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aimagelab/meshed-memory-transformer/HEAD/vocab.pkl --------------------------------------------------------------------------------