Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"
☆104Oct 3, 2023Updated 2 years ago
Alternatives and similar repositories for SPACE
Users that are interested in SPACE are comparing it to the libraries listed below
Sorting:
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Dec 8, 2022Updated 3 years ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Sep 3, 2022Updated 3 years ago
- Official Release of ICLR 2020 paper "SCALOR: Generative World Models with Scalable Object Representations"☆49Dec 24, 2023Updated 2 years ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆281Updated this week
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆59May 16, 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
- Code, data and benchmark from the paper "Unmasking the Inductive Biases of Unsupervised Object Representations for Video Sequences".☆36Sep 7, 2021Updated 4 years ago
- EfficientMORL (ICML'21)☆22Nov 3, 2021Updated 4 years ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- A toolbox of compositional scene representation learning methods and benchmark datasets.☆12Mar 2, 2024Updated 2 years ago
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Feb 27, 2023Updated 3 years ago
- Official code for Slot-Transformer for Videos (STEVE)☆50Jan 9, 2023Updated 3 years ago
- Official Release of NeurIPS 2020 Spotlight paper "Generative Neurosymbolic Machines"☆36Mar 9, 2024Updated last year
- Original tensorflow implementation of SILOT (Spatially Invariant, Label-free Object Tracking).☆13Mar 24, 2023Updated 2 years ago
- ☆22Sep 22, 2022Updated 3 years ago
- ☆42Jan 22, 2024Updated 2 years ago
- Unsupervised Part Discovery from Contrastive Reconstruction (NeurIPS 2021)☆67Dec 6, 2023Updated 2 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- Implementation of Sum-Product Attend-Infer-Repeat☆31May 21, 2020Updated 5 years ago
- Library for Auto-Encoding Sequential Monte Carlo☆18Jan 29, 2024Updated 2 years ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 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 …☆88Jan 9, 2023Updated 3 years ago
- Code for 'Emergent Symbols through Binding in External Memory'.☆21May 4, 2023Updated 2 years ago
- This repository is the official implementation of Improving Object-centric Learning With Query Optimization☆51May 30, 2023Updated 2 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆47Mar 24, 2023Updated 2 years ago
- Official Code for Neural Systematic Binder☆34Mar 27, 2023Updated 2 years ago
- ☆72Jun 20, 2022Updated 3 years ago
- Learning Long-term Visual Dynamics with Region Proposal Interaction Networks (ICLR 2021)☆113May 29, 2022Updated 3 years ago
- Variational Autoencoder with Spatial Broadcast Decoder☆35Jul 12, 2019Updated 6 years ago
- [NeurIPS 2021] Unsupervised Foreground Extraction via Deep Region Competition☆43Updated this week
- Compositional Object Light Fields code☆27Oct 9, 2022Updated 3 years ago
- When can you tell whether an image has been cropped or not?☆29Sep 19, 2021Updated 4 years ago
- Official code for the paper: "Metadata Archaeology"☆19May 10, 2023Updated 2 years ago
- PyTorch Implementation for the paper "C3VQG: Category Consistent Cyclic Visual Question Generation" (ACM MM Asia'20).☆17Mar 31, 2023Updated 2 years ago
- TensorFlow-based implementation of "Attend, Infer, Repeat" paper (Eslami et al., 2016, arXiv:1603.08575).☆42Oct 29, 2017Updated 8 years ago
- Implementation of Sequential Attend, Infer, Repeat (SQAIR)☆96Apr 9, 2019Updated 6 years ago
- Official Repository of NeurIPS2021 paper: PTR☆32Dec 17, 2021Updated 4 years ago
- Original implementation of Spatially Invariant Attend, Infer, Repeat (SPAIR) in TensorFlow.☆51Jun 18, 2020Updated 5 years ago