├── .gitignore ├── CITATION.cff ├── LICENSE ├── README.md ├── data_generation ├── data_generation_causal_world.py ├── data_generation_ithor.py └── data_generation_voronoi.py ├── demo.ipynb ├── docs ├── index.html └── static │ ├── css │ ├── bulma-carousel.min.css │ ├── bulma-slider.min.css │ ├── bulma.css.map.txt │ ├── bulma.min.css │ ├── fontawesome.all.min.css │ └── index.css │ ├── images │ ├── banner_image.png │ ├── favicon.ico │ ├── interaction_map_1.png │ ├── interaction_map_2.png │ ├── interaction_map_3.png │ ├── interaction_map_4.png │ ├── teaser.png │ ├── triplet_1.png │ ├── triplet_2.png │ ├── triplet_3.png │ ├── triplet_4.png │ ├── triplet_5.png │ └── triplet_generation.png │ ├── js │ ├── bulma-carousel.js │ ├── bulma-carousel.min.js │ ├── bulma-slider.js │ ├── bulma-slider.min.js │ ├── fontawesome.all.min.js │ └── index.js │ ├── pdfs │ └── UAI_Poster.pdf │ └── videos │ ├── banner_video.mov │ ├── causal_world_seq.mov │ └── ithor_seq.mov ├── environment.yml ├── experiments ├── datasets.py ├── train_ae.py ├── train_nf.py ├── train_vae.py └── utils.py └── models ├── ae ├── __init__.py └── lightning_module.py ├── biscuit_nf ├── __init__.py └── lightning_module.py ├── biscuit_vae ├── __init__.py └── lightning_module.py └── shared ├── __init__.py ├── callbacks.py ├── causal_encoder.py ├── encoder_decoder.py ├── flow_layers.py ├── modules.py ├── transition_prior.py ├── utils.py └── visualization.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/README.md -------------------------------------------------------------------------------- /data_generation/data_generation_causal_world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/data_generation/data_generation_causal_world.py -------------------------------------------------------------------------------- /data_generation/data_generation_ithor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/data_generation/data_generation_ithor.py -------------------------------------------------------------------------------- /data_generation/data_generation_voronoi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/data_generation/data_generation_voronoi.py -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/demo.ipynb -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/static/css/bulma-carousel.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/css/bulma-carousel.min.css -------------------------------------------------------------------------------- /docs/static/css/bulma-slider.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/css/bulma-slider.min.css -------------------------------------------------------------------------------- /docs/static/css/bulma.css.map.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/css/bulma.css.map.txt -------------------------------------------------------------------------------- /docs/static/css/bulma.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/css/bulma.min.css -------------------------------------------------------------------------------- /docs/static/css/fontawesome.all.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/css/fontawesome.all.min.css -------------------------------------------------------------------------------- /docs/static/css/index.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/css/index.css -------------------------------------------------------------------------------- /docs/static/images/banner_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/banner_image.png -------------------------------------------------------------------------------- /docs/static/images/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/favicon.ico -------------------------------------------------------------------------------- /docs/static/images/interaction_map_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/interaction_map_1.png -------------------------------------------------------------------------------- /docs/static/images/interaction_map_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/interaction_map_2.png -------------------------------------------------------------------------------- /docs/static/images/interaction_map_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/interaction_map_3.png -------------------------------------------------------------------------------- /docs/static/images/interaction_map_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/interaction_map_4.png -------------------------------------------------------------------------------- /docs/static/images/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/teaser.png -------------------------------------------------------------------------------- /docs/static/images/triplet_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/triplet_1.png -------------------------------------------------------------------------------- /docs/static/images/triplet_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/triplet_2.png -------------------------------------------------------------------------------- /docs/static/images/triplet_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/triplet_3.png -------------------------------------------------------------------------------- /docs/static/images/triplet_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/triplet_4.png -------------------------------------------------------------------------------- /docs/static/images/triplet_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/triplet_5.png -------------------------------------------------------------------------------- /docs/static/images/triplet_generation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/images/triplet_generation.png -------------------------------------------------------------------------------- /docs/static/js/bulma-carousel.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/js/bulma-carousel.js -------------------------------------------------------------------------------- /docs/static/js/bulma-carousel.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/js/bulma-carousel.min.js -------------------------------------------------------------------------------- /docs/static/js/bulma-slider.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/js/bulma-slider.js -------------------------------------------------------------------------------- /docs/static/js/bulma-slider.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/js/bulma-slider.min.js -------------------------------------------------------------------------------- /docs/static/js/fontawesome.all.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/js/fontawesome.all.min.js -------------------------------------------------------------------------------- /docs/static/js/index.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/js/index.js -------------------------------------------------------------------------------- /docs/static/pdfs/UAI_Poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/pdfs/UAI_Poster.pdf -------------------------------------------------------------------------------- /docs/static/videos/banner_video.mov: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/videos/banner_video.mov -------------------------------------------------------------------------------- /docs/static/videos/causal_world_seq.mov: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/videos/causal_world_seq.mov -------------------------------------------------------------------------------- /docs/static/videos/ithor_seq.mov: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/docs/static/videos/ithor_seq.mov -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/environment.yml -------------------------------------------------------------------------------- /experiments/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/experiments/datasets.py -------------------------------------------------------------------------------- /experiments/train_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/experiments/train_ae.py -------------------------------------------------------------------------------- /experiments/train_nf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/experiments/train_nf.py -------------------------------------------------------------------------------- /experiments/train_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/experiments/train_vae.py -------------------------------------------------------------------------------- /experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/experiments/utils.py -------------------------------------------------------------------------------- /models/ae/__init__.py: -------------------------------------------------------------------------------- 1 | from models.ae.lightning_module import Autoencoder -------------------------------------------------------------------------------- /models/ae/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/ae/lightning_module.py -------------------------------------------------------------------------------- /models/biscuit_nf/__init__.py: -------------------------------------------------------------------------------- 1 | from models.biscuit_nf.lightning_module import BISCUITNF -------------------------------------------------------------------------------- /models/biscuit_nf/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/biscuit_nf/lightning_module.py -------------------------------------------------------------------------------- /models/biscuit_vae/__init__.py: -------------------------------------------------------------------------------- 1 | from models.biscuit_vae.lightning_module import BISCUITVAE -------------------------------------------------------------------------------- /models/biscuit_vae/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/biscuit_vae/lightning_module.py -------------------------------------------------------------------------------- /models/shared/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/shared/__init__.py -------------------------------------------------------------------------------- /models/shared/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/shared/callbacks.py -------------------------------------------------------------------------------- /models/shared/causal_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/shared/causal_encoder.py -------------------------------------------------------------------------------- /models/shared/encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/shared/encoder_decoder.py -------------------------------------------------------------------------------- /models/shared/flow_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/shared/flow_layers.py -------------------------------------------------------------------------------- /models/shared/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/shared/modules.py -------------------------------------------------------------------------------- /models/shared/transition_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/shared/transition_prior.py -------------------------------------------------------------------------------- /models/shared/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/shared/utils.py -------------------------------------------------------------------------------- /models/shared/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/BISCUIT/HEAD/models/shared/visualization.py --------------------------------------------------------------------------------