zhixuan-lin / G-SWMView external linksLinks
Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"
☆37Dec 8, 2022Updated 3 years ago
Alternatives and similar repositories for G-SWM
Users that are interested in G-SWM are comparing it to the libraries listed below
Sorting:
- Official Release of ICLR 2020 paper "SCALOR: Generative World Models with Scalable Object Representations"☆49Dec 24, 2023Updated 2 years ago
- Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"☆104Oct 3, 2023Updated 2 years ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- Official code for Slot-Transformer for Videos (STEVE)☆50Jan 9, 2023Updated 3 years ago
- [CVPR 2020] A generative model with latent factors that are independent and localized.☆12Mar 27, 2025Updated 10 months ago
- Original tensorflow implementation of SILOT (Spatially Invariant, Label-free Object Tracking).☆13Mar 24, 2023Updated 2 years ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Sep 3, 2022Updated 3 years ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 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
- An official implementation of the paper "Unsupervised Keypoint Learning for Guiding Class-conditional Video Prediction", NeurIPS 2019☆30Nov 22, 2022Updated 3 years ago
- ☆42Jan 22, 2024Updated 2 years ago
- ☆22Sep 22, 2022Updated 3 years ago
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆59May 16, 2022Updated 3 years ago
- PALMER: Perception-Action Loop with Memory for Long-Horizon Planning, NeurIPS 2022☆15Dec 12, 2022Updated 3 years ago
- Official PyTorch implementation of GENESIS and GENESIS-V2☆109Apr 13, 2022Updated 3 years ago
- Burgess et al. "MONet: Unsupervised Scene Decomposition and Representation"☆89Dec 3, 2022Updated 3 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 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
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆281Dec 17, 2021Updated 4 years ago
- ☆88Aug 13, 2025Updated 6 months ago
- EfficientMORL (ICML'21)☆22Nov 3, 2021Updated 4 years ago
- Repo for our ICML21 paper, Unsupervised Learning of Visual 3D Keypoints for Control☆46Feb 17, 2025Updated 11 months ago
- Unsupervised Discovery of Object Radiance Fields☆174Jan 13, 2023Updated 3 years ago
- Lipschitz Lifelong RL☆11Nov 6, 2020Updated 5 years ago
- [ICML2023] Instant Soup Cheap Pruning Ensembles in A Single Pass Can Draw Lottery Tickets from Large Models. Ajay Jaiswal, Shiwei Liu, Ti…☆11Nov 28, 2023Updated 2 years ago
- Progressive Growing of Points with Tree-structured Generators (BMVC 2021)☆11Nov 1, 2023Updated 2 years ago
- Official implementation for GATSBI: Generative Agent-centric Spatio-temporal Object Interaction (CVPR'2021)☆12Mar 23, 2022Updated 3 years ago
- Library for the training and evaluation of object-centric models (ICML 2022)☆71Apr 30, 2023Updated 2 years ago
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆12Feb 23, 2024Updated last year
- Official repository for "Temporal Disentanglement of Representations for Improved Generalisation in Reinforcement Learning".☆13Jan 25, 2023Updated 3 years ago
- Tensorflow models and simulation code for 'ShapeStacks: Learning Vision-Based Physical Intuition for Generalised Object Stacking'☆46Mar 24, 2023Updated 2 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- ☆44Feb 12, 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
- Deep Hierarchical Planning from Pixels☆114Dec 21, 2022Updated 3 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆163Jun 23, 2023Updated 2 years ago
- Repository for the paper "Long-Horizon Visual Planning with Goal-Conditioned Hierarchical Predictors"☆46Nov 22, 2022Updated 3 years ago
- Original implementation of Spatially Invariant Attend, Infer, Repeat (SPAIR) in TensorFlow.☆51Jun 18, 2020Updated 5 years ago
- Official Github repository for Dynamic Mesh Recovery from Partial Point Cloud Sequence (ICCV 2023)☆13Sep 13, 2024Updated last year