Multi-object image datasets with ground-truth segmentation masks and generative factors.
☆284Mar 3, 2026Updated 2 weeks ago
Alternatives and similar repositories for multi_object_datasets
Users that are interested in multi_object_datasets are comparing it to the libraries listed below
Sorting:
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆59May 16, 2022Updated 3 years ago
- Burgess et al. "MONet: Unsupervised Scene Decomposition and Representation"☆89Dec 3, 2022Updated 3 years ago
- Official PyTorch implementation of GENESIS and GENESIS-V2☆109Apr 13, 2022Updated 3 years ago
- Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"☆104Oct 3, 2023Updated 2 years ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Sep 3, 2022Updated 3 years ago
- This is the official source code for SLATE. We provide the code for the model, the training code, and a dataset loader for the 3D Shapes …☆88Jan 9, 2023Updated 3 years ago
- Code, data and benchmark from the paper "Unmasking the Inductive Biases of Unsupervised Object Representations for Video Sequences".☆36Sep 7, 2021Updated 4 years ago
- ☆42Jan 22, 2024Updated 2 years ago
- Official Release of ICLR 2020 paper "SCALOR: Generative World Models with Scalable Object Representations"☆49Dec 24, 2023Updated 2 years ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Dec 8, 2022Updated 3 years ago
- A Diagnostic Dataset for Compositional Language and Elementary Visual Reasoning☆644Aug 30, 2021Updated 4 years ago
- EfficientMORL (ICML'21)☆22Nov 3, 2021Updated 4 years ago
- ☆89Aug 13, 2025Updated 7 months ago
- Official Code for Neural Systematic Binder☆34Mar 27, 2023Updated 2 years ago
- Code for the paper "Generative Modeling of Infinite Occluded Objects for Compositional Scene Representation"☆10Feb 4, 2023Updated 3 years ago
- Code for the "Binding via Reconstruction Clustering" paper☆21Jan 19, 2016Updated 10 years ago
- Official Release of NeurIPS 2020 Spotlight paper "Generative Neurosymbolic Machines"☆36Mar 9, 2024Updated 2 years ago
- ☆180Feb 3, 2023Updated 3 years ago
- CATER: A diagnostic dataset for Compositional Actions and TEmporal Reasoning☆108Dec 18, 2020Updated 5 years ago
- Library for the training and evaluation of object-centric models (ICML 2022)☆71Apr 30, 2023Updated 2 years ago
- Contrastive Learning of Structured World Models☆396Jun 3, 2020Updated 5 years ago
- Dataset to assess the disentanglement properties of unsupervised learning methods☆531Jan 3, 2021Updated 5 years ago
- Official code for Slot-Transformer for Videos (STEVE)☆51Jan 9, 2023Updated 3 years ago
- Official PyTorch implementation of 'RELATE: Physically Plausible Multi-Object SceneSynthesis Using Structured Latent Spaces'.☆31Nov 6, 2020Updated 5 years ago
- Original implementation of Spatially Invariant Attend, Infer, Repeat (SPAIR) in TensorFlow.☆51Jun 18, 2020Updated 5 years ago
- Spriteworld: a flexible, configurable python-based reinforcement learning environment☆371Jun 1, 2020Updated 5 years ago
- Unsupervised Discovery of Object Radiance Fields☆175Jan 13, 2023Updated 3 years ago
- Code from the article: "The Role of Disentanglement in Generalisation" (ICLR, 2021).☆21May 28, 2022Updated 3 years ago
- Code release for NeurIPS 2023 paper SlotDiffusion: Object-centric Learning with Diffusion Models☆94Jan 16, 2024Updated 2 years ago
- A toolbox of compositional scene representation learning methods and benchmark datasets.☆12Mar 2, 2024Updated 2 years ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- Code for the "Relational Neural Expectation Maximization: Unsupervised Discovery of Objects and their Interactions" paper.☆73Nov 22, 2022Updated 3 years ago
- Compositional Object Light Fields code☆27Oct 9, 2022Updated 3 years ago
- Code release for ICLR 2023 paper: SlotFormer on object-centric dynamics models☆120Sep 20, 2023Updated 2 years ago
- A data generation pipeline for creating semi-realistic synthetic multi-object videos with rich annotations such as instance segmentation …☆2,679May 6, 2025Updated 10 months ago
- PyTorch implementation of paper "Visual Concept-Metaconcept Learner", NeruIPS 2019☆47Dec 3, 2019Updated 6 years ago
- Official Release of NeurIPS 2023 Spotlight paper "Object-Centric Slot Diffusion"☆72Mar 9, 2024Updated 2 years ago
- ☆22Sep 22, 2022Updated 3 years ago
- PyTorch implementation for the Neuro-Symbolic Concept Learner (NS-CL).☆451Oct 24, 2020Updated 5 years ago