zhixuan-lin / SPACELinks
Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"
☆104Updated 2 years ago
Alternatives and similar repositories for SPACE
Users that are interested in SPACE are comparing it to the libraries listed below
Sorting:
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆59Updated 3 years ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Updated 2 years ago
- Official Release of ICLR 2020 paper "SCALOR: Generative World Models with Scalable Object Representations"☆49Updated last year
- Official PyTorch implementation of GENESIS and GENESIS-V2☆110Updated 3 years ago
- Burgess et al. "MONet: Unsupervised Scene Decomposition and Representation"☆89Updated 2 years ago
- Code, data and benchmark from the paper "Unmasking the Inductive Biases of Unsupervised Object Representations for Video Sequences".☆36Updated 4 years ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Updated 3 years ago
- EfficientMORL (ICML'21)☆21Updated 3 years ago
- Library for the training and evaluation of object-centric models (ICML 2022)☆70Updated 2 years ago
- "CoPhy: Counterfactual Learning of Physical Dynamics", F. Baradel, N. Neverova, J. Mille, G. Mori, C. Wolf, ICLR'2020☆34Updated 5 years ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Updated 4 years ago
- Structured Object-Aware Physics Prediction for Video Modeling and Planning☆32Updated 5 years ago
- Learning Long-term Visual Dynamics with Region Proposal Interaction Networks (ICLR 2021)☆113Updated 3 years ago
- Original tensorflow implementation of SILOT (Spatially Invariant, Label-free Object Tracking).☆13Updated 2 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 …☆86Updated 2 years ago
- Neural Networks that convert input movies into Physical Scene Graphs (PSGs)☆63Updated 4 years ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆275Updated 3 years ago
- Transporter implementation in PyTorch☆20Updated 6 years ago
- Reinforcement Learning with Latent Flow☆44Updated 4 years ago
- Official Release of NeurIPS 2020 Spotlight paper "Generative Neurosymbolic Machines"☆35Updated last year
- Clockwork Variational Autoencoder☆48Updated 4 years ago
- Training code and evaluation benchmarks for the "Self-Supervised Policy Adaptation during Deployment" paper.☆113Updated 4 years ago
- Variational Autoencoder with Spatial Broadcast Decoder☆35Updated 6 years ago
- ☆42Updated last year
- Code for the "Relational Neural Expectation Maximization: Unsupervised Discovery of Objects and their Interactions" paper.☆73Updated 2 years ago
- Unsupervised learning of Object Landmarks through Conditional Image Generation☆48Updated 5 years ago
- ☆79Updated 5 months ago
- ☆66Updated 4 years ago
- [ICML 2019] TensorFlow Code for Self-Supervised Exploration via Disagreement☆128Updated 6 years ago
- Code for image generation of Variational Discriminator Bottleneck: Improving Imitation Learning, Inverse RL, and GANs by Constraining Inf…☆144Updated 2 years ago