google-deepmind / multi_object_datasets
Multi-object image datasets with ground-truth segmentation masks and generative factors.
☆264Updated 3 years 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
- Burgess et al. "MONet: Unsupervised Scene Decomposition and Representation"☆88Updated 2 years ago
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆58Updated 2 years ago
- Library for the training and evaluation of object-centric models (ICML 2022)☆68Updated last year
- Contrastive Learning of Structured World Models☆390Updated 4 years ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Updated 2 years ago
- Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"☆102Updated last year
- This repository contains the 3D shapes dataset, used in Kim, Hyunjik and Mnih, Andriy. "Disentangling by Factorising." In Proceedings of …☆140Updated last year
- Official PyTorch implementation of GENESIS and GENESIS-V2☆111Updated 2 years ago
- Generative Query Network (GQN) in PyTorch as described in "Neural Scene Representation and Rendering"☆320Updated 5 years ago
- PyTorch Implementation of Generative Query Network☆138Updated 6 years ago
- PHYRE is a benchmark for physical reasoning.☆440Updated last year
- Self-supervised learning through the eyes of a child☆139Updated 3 years ago
- CATER: A diagnostic dataset for Compositional Actions and TEmporal Reasoning☆104Updated 4 years ago
- A Diagnostic Dataset for Compositional Language and Elementary Visual Reasoning☆606Updated 3 years ago
- Code, data and benchmark from the paper "Unmasking the Inductive Biases of Unsupervised Object Representations for Video Sequences".☆36Updated 3 years ago
- Implementation of Slot Attention from GoogleAI☆417Updated 7 months ago
- Dataset to assess the disentanglement properties of unsupervised learning methods☆496Updated 4 years ago
- Learning Long-term Visual Dynamics with Region Proposal Interaction Networks (ICLR 2021)☆112Updated 2 years ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆35Updated 2 years ago
- Neural-symbolic visual question answering☆263Updated last year
- EfficientMORL (ICML'21)☆22Updated 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 …☆84Updated 2 years ago
- Repository for the paper "Planning to Explore via Self-Supervised World Models"☆218Updated 2 years ago
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆193Updated 3 years ago
- PyTorch implementation of ICLR 2020 paper "CLEVRER: CoLlision Events for Video REpresentation and Reasoning"☆115Updated 4 years ago
- Code for "Unsupervised State Representation Learning in Atari"☆247Updated last year
- Code for the "Relational Neural Expectation Maximization: Unsupervised Discovery of Objects and their Interactions" paper.☆73Updated 2 years ago
- Code release for ICLR 2023 paper: SlotFormer on object-centric dynamics models☆106Updated last year
- Measuring compositionality in representation learning☆72Updated 5 years ago
- Official Release of ICLR 2020 paper "SCALOR: Generative World Models with Scalable Object Representations"☆49Updated last year