├── .gitignore ├── README.md ├── accelerate_config.yaml ├── assets ├── ade20_decomp.png ├── ade20k_training_daam.png ├── art_decomposition.png ├── demo.gif ├── imagenet_decomposition.png ├── supp_ade20k_decomposition.png ├── supp_external_composition.png └── teaser.png ├── compose_models.py ├── create_accelerate_config.py ├── daam_ddim_visualize.py ├── daam_visualize_generation.py ├── datasets.py ├── eval.py ├── inference.py ├── main.py ├── requirements.txt └── scripts ├── evaluate_classification.sh ├── sample.sh ├── train_ADE20K.sh ├── train_art.sh └── train_imagenet.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/README.md -------------------------------------------------------------------------------- /accelerate_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/accelerate_config.yaml -------------------------------------------------------------------------------- /assets/ade20_decomp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/assets/ade20_decomp.png -------------------------------------------------------------------------------- /assets/ade20k_training_daam.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/assets/ade20k_training_daam.png -------------------------------------------------------------------------------- /assets/art_decomposition.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/assets/art_decomposition.png -------------------------------------------------------------------------------- /assets/demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/assets/demo.gif -------------------------------------------------------------------------------- /assets/imagenet_decomposition.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/assets/imagenet_decomposition.png -------------------------------------------------------------------------------- /assets/supp_ade20k_decomposition.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/assets/supp_ade20k_decomposition.png -------------------------------------------------------------------------------- /assets/supp_external_composition.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/assets/supp_external_composition.png -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /compose_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/compose_models.py -------------------------------------------------------------------------------- /create_accelerate_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/create_accelerate_config.py -------------------------------------------------------------------------------- /daam_ddim_visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/daam_ddim_visualize.py -------------------------------------------------------------------------------- /daam_visualize_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/daam_visualize_generation.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/datasets.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/eval.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/inference.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/evaluate_classification.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/scripts/evaluate_classification.sh -------------------------------------------------------------------------------- /scripts/sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/scripts/sample.sh -------------------------------------------------------------------------------- /scripts/train_ADE20K.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/scripts/train_ADE20K.sh -------------------------------------------------------------------------------- /scripts/train_art.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/scripts/train_art.sh -------------------------------------------------------------------------------- /scripts/train_imagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nanlliu/Unsupervised-Compositional-Concepts-Discovery/HEAD/scripts/train_imagenet.sh --------------------------------------------------------------------------------