zhixuan-lin / G-SWM
Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"
☆35Updated 2 years ago
Alternatives and similar repositories for G-SWM:
Users that are interested in G-SWM are comparing it to the libraries listed below
- Code, data and benchmark from the paper "Unmasking the Inductive Biases of Unsupervised Object Representations for Video Sequences".☆36Updated 3 years ago
- EfficientMORL (ICML'21)☆22Updated 3 years ago
- Library for the training and evaluation of object-centric models (ICML 2022)☆66Updated last year
- Official Release of ICLR 2020 paper "SCALOR: Generative World Models with Scalable Object Representations"☆48Updated last year
- "CoPhy: Counterfactual Learning of Physical Dynamics", F. Baradel, N. Neverova, J. Mille, G. Mori, C. Wolf, ICLR'2020☆33Updated 4 years ago
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆58Updated 2 years ago
- Structured Object-Aware Physics Prediction for Video Modeling and Planning☆32Updated 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 …☆84Updated 2 years ago
- Original tensorflow implementation of SILOT (Spatially Invariant, Label-free Object Tracking).☆13Updated last year
- ☆40Updated last year
- Official Code for Neural Systematic Binder☆30Updated last year
- Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"☆102Updated last year
- Official code for Slot-Transformer for Videos (STEVE)☆49Updated 2 years ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆56Updated 4 years ago
- Clockwork Variational Autoencoder☆46Updated 3 years ago
- ☆22Updated 2 years ago
- Transporter implementation in PyTorch☆20Updated 5 years ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Updated 2 years ago
- Official Release of NeurIPS 2020 Spotlight paper "Generative Neurosymbolic Machines"☆35Updated 10 months ago
- Code release for ICLR 2023 paper: SlotFormer on object-centric dynamics models☆104Updated last year
- Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…☆9Updated 3 weeks ago
- ☆59Updated 6 months ago
- Solving reinforcement learning tasks which require language and vision☆32Updated last year
- Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)☆20Updated 3 years ago
- Neural Networks that convert input movies into Physical Scene Graphs (PSGs)☆62Updated 3 years ago
- Latent World Models For Intrinsically Motivated Exploration | Official repository☆22Updated 3 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 4 years ago
- [NeurIPS 2022] Compositional Generalization in Unsupervised Compositional Representation Learning: A Study on Disentanglement and Emergen…☆13Updated 2 years ago
- ☆54Updated 3 years ago
- Learning Long-term Visual Dynamics with Region Proposal Interaction Networks (ICLR 2021)☆112Updated 2 years ago