zhixuan-lin / SPACE
Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"
☆101Updated last year
Related projects ⓘ
Alternatives and complementary repositories for SPACE
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆58Updated 2 years ago
- Code, data and benchmark from the paper "Unmasking the Inductive Biases of Unsupervised Object Representations for Video Sequences".☆36Updated 3 years ago
- Official PyTorch implementation of GENESIS and GENESIS-V2☆97Updated 2 years ago
- Burgess et al. "MONet: Unsupervised Scene Decomposition and Representation"☆89Updated last year
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆34Updated last year
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆58Updated 2 years ago
- EfficientMORL (ICML'21)☆22Updated 3 years ago
- Learning Long-term Visual Dynamics with Region Proposal Interaction Networks (ICLR 2021)☆112Updated 2 years ago
- Library for the training and evaluation of object-centric models (ICML 2022)☆65Updated last year
- Official Release of ICLR 2020 paper "SCALOR: Generative World Models with Scalable Object Representations"☆48Updated 10 months ago
- "CoPhy: Counterfactual Learning of Physical Dynamics", F. Baradel, N. Neverova, J. Mille, G. Mori, C. Wolf, ICLR'2020☆33Updated 4 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 …☆82Updated last year
- ☆39Updated 9 months ago
- Learning Spatial Common Sense with Geometry-Aware Recurrent Networks☆55Updated 4 years ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆55Updated 3 years ago
- CATER: A diagnostic dataset for Compositional Actions and TEmporal Reasoning☆103Updated 3 years ago
- Official Release of NeurIPS 2020 Spotlight paper "Generative Neurosymbolic Machines"☆35Updated 8 months ago
- Temporal Difference Variational Auto-Encoder (TD-VAE) (Implementation in PyTorch)☆154Updated last year
- Structured Object-Aware Physics Prediction for Video Modeling and Planning☆32Updated 4 years ago
- Clockwork Variational Autoencoder☆45Updated 3 years ago
- ☆73Updated 6 months ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆257Updated 2 years ago
- Original tensorflow implementation of SILOT (Spatially Invariant, Label-free Object Tracking).☆13Updated last year
- Implicit Generation and Generalization in Energy Based Models in PyTorch☆65Updated 5 years ago
- Variational Autoencoder with Spatial Broadcast Decoder☆35Updated 5 years ago
- cordial-sync is a software package than can be used to reproduce the results from the paper "A Cordial Sync: Going Beyond Marginal Polici…☆37Updated 3 years ago
- Code for image generation of Variational Discriminator Bottleneck: Improving Imitation Learning, Inverse RL, and GANs by Constraining Inf…☆141Updated last year
- Implementation of IODINE model☆10Updated 5 years ago
- [ICLR 2019] ]Unsupervised Discovery of Parts, Structure, and Dynamics☆46Updated last year