├── .gitignore ├── LICENSE ├── README.md ├── aao ├── 193-annot.csv └── evaluate_att_att_obj.py ├── assets └── csp-figure.png ├── clip_modules ├── __init__.py ├── interface.py ├── model_loader.py └── text_encoder.py ├── credits.md ├── datasets ├── __init__.py ├── composition_dataset.py ├── feasibility.py ├── read_datasets.py └── reorganize_utzap.py ├── download_data.sh ├── evaluate.py ├── mix ├── __init__.py ├── evaluate_mix_train.py └── mix_train.py ├── models ├── __init__.py ├── compositional_modules.py ├── coop.py └── csp.py ├── requirements.txt ├── train.py └── utils ├── __init__.py └── core.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/README.md -------------------------------------------------------------------------------- /aao/193-annot.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/aao/193-annot.csv -------------------------------------------------------------------------------- /aao/evaluate_att_att_obj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/aao/evaluate_att_att_obj.py -------------------------------------------------------------------------------- /assets/csp-figure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/assets/csp-figure.png -------------------------------------------------------------------------------- /clip_modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/clip_modules/__init__.py -------------------------------------------------------------------------------- /clip_modules/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/clip_modules/interface.py -------------------------------------------------------------------------------- /clip_modules/model_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/clip_modules/model_loader.py -------------------------------------------------------------------------------- /clip_modules/text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/clip_modules/text_encoder.py -------------------------------------------------------------------------------- /credits.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/credits.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/composition_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/datasets/composition_dataset.py -------------------------------------------------------------------------------- /datasets/feasibility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/datasets/feasibility.py -------------------------------------------------------------------------------- /datasets/read_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/datasets/read_datasets.py -------------------------------------------------------------------------------- /datasets/reorganize_utzap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/datasets/reorganize_utzap.py -------------------------------------------------------------------------------- /download_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/download_data.sh -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/evaluate.py -------------------------------------------------------------------------------- /mix/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mix/evaluate_mix_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/mix/evaluate_mix_train.py -------------------------------------------------------------------------------- /mix/mix_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/mix/mix_train.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/compositional_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/models/compositional_modules.py -------------------------------------------------------------------------------- /models/coop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/models/coop.py -------------------------------------------------------------------------------- /models/csp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/models/csp.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | from utils.core import set_seed -------------------------------------------------------------------------------- /utils/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BatsResearch/csp/HEAD/utils/core.py --------------------------------------------------------------------------------