├── readme.md └── src ├── config.py ├── main ├── __init__.py ├── caption │ ├── model.py │ ├── model1.py │ └── model2.py ├── data.py ├── evaluate.py └── train_mem_sg.py ├── scripts ├── __init__.py └── parse_sg.py ├── styled_eval ├── __init__.py ├── clf.py ├── clf_cnn.py ├── eval.py ├── lm_kenlm.py ├── lm_srilm.py └── train_clf_nn_chn.py └── util ├── __init__.py ├── customjson.py ├── data.py ├── evaluate.py ├── loss.py ├── model.py ├── pipeline.py ├── preprocess.py ├── reward ├── __init__.py ├── bleu │ ├── __init__.py │ ├── bleu.py │ └── bleu_scorer.py ├── ciderD │ ├── __init__.py │ ├── ciderD.py │ └── ciderD_scorer.py └── style_reward.py └── vocabulary.py /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/readme.md -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/config.py -------------------------------------------------------------------------------- /src/main/__init__.py: -------------------------------------------------------------------------------- 1 | # --- -------------------------------------------------------------------------------- /src/main/caption/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/main/caption/model.py -------------------------------------------------------------------------------- /src/main/caption/model1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/main/caption/model1.py -------------------------------------------------------------------------------- /src/main/caption/model2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/main/caption/model2.py -------------------------------------------------------------------------------- /src/main/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/main/data.py -------------------------------------------------------------------------------- /src/main/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/main/evaluate.py -------------------------------------------------------------------------------- /src/main/train_mem_sg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/main/train_mem_sg.py -------------------------------------------------------------------------------- /src/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | # -------------------------------------------------------------------------------- /src/scripts/parse_sg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/scripts/parse_sg.py -------------------------------------------------------------------------------- /src/styled_eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/styled_eval/__init__.py -------------------------------------------------------------------------------- /src/styled_eval/clf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/styled_eval/clf.py -------------------------------------------------------------------------------- /src/styled_eval/clf_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/styled_eval/clf_cnn.py -------------------------------------------------------------------------------- /src/styled_eval/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/styled_eval/eval.py -------------------------------------------------------------------------------- /src/styled_eval/lm_kenlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/styled_eval/lm_kenlm.py -------------------------------------------------------------------------------- /src/styled_eval/lm_srilm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/styled_eval/lm_srilm.py -------------------------------------------------------------------------------- /src/styled_eval/train_clf_nn_chn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/styled_eval/train_clf_nn_chn.py -------------------------------------------------------------------------------- /src/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/__init__.py -------------------------------------------------------------------------------- /src/util/customjson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/customjson.py -------------------------------------------------------------------------------- /src/util/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/data.py -------------------------------------------------------------------------------- /src/util/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/evaluate.py -------------------------------------------------------------------------------- /src/util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/loss.py -------------------------------------------------------------------------------- /src/util/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/model.py -------------------------------------------------------------------------------- /src/util/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/pipeline.py -------------------------------------------------------------------------------- /src/util/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/preprocess.py -------------------------------------------------------------------------------- /src/util/reward/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/reward/__init__.py -------------------------------------------------------------------------------- /src/util/reward/bleu/__init__.py: -------------------------------------------------------------------------------- 1 | # Bleu from https://github.com/ruotianluo/coco-caption -------------------------------------------------------------------------------- /src/util/reward/bleu/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/reward/bleu/bleu.py -------------------------------------------------------------------------------- /src/util/reward/bleu/bleu_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/reward/bleu/bleu_scorer.py -------------------------------------------------------------------------------- /src/util/reward/ciderD/__init__.py: -------------------------------------------------------------------------------- 1 | # CIDEr-D code from https://github.com/ruotianluo/cider -------------------------------------------------------------------------------- /src/util/reward/ciderD/ciderD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/reward/ciderD/ciderD.py -------------------------------------------------------------------------------- /src/util/reward/ciderD/ciderD_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/reward/ciderD/ciderD_scorer.py -------------------------------------------------------------------------------- /src/util/reward/style_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/reward/style_reward.py -------------------------------------------------------------------------------- /src/util/vocabulary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/entalent/MemCap/HEAD/src/util/vocabulary.py --------------------------------------------------------------------------------