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 dataset. This code is implemented in Pytorch.
☆88Jan 9, 2023Updated 3 years ago
Alternatives and similar repositories for slate
Users that are interested in slate are comparing it to the libraries listed below
Sorting:
- ☆89Aug 13, 2025Updated 7 months ago
- ☆42Jan 22, 2024Updated 2 years ago
- Official code for Slot-Transformer for Videos (STEVE)☆51Jan 9, 2023Updated 3 years ago
- Code release for ICLR 2023 paper: SlotFormer on object-centric dynamics models☆120Sep 20, 2023Updated 2 years ago
- ☆180Feb 3, 2023Updated 3 years ago
- Library for the training and evaluation of object-centric models (ICML 2022)☆71Apr 30, 2023Updated 2 years ago
- This repository is the official implementation of Improving Object-centric Learning With Query Optimization☆51May 30, 2023Updated 2 years ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆284Mar 3, 2026Updated 2 weeks ago
- [CVPR 2024 Highlight] SPOT: Self-Training with Patch-Order Permutation for Object-Centric Learning with Autoregressive Transformers☆73Jun 11, 2024Updated last year
- Official Release of NeurIPS 2020 Spotlight paper "Generative Neurosymbolic Machines"☆36Mar 9, 2024Updated 2 years ago
- Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"☆37Dec 8, 2022Updated 3 years ago
- Official Code for Neural Systematic Binder☆34Mar 27, 2023Updated 2 years ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆62Sep 21, 2022Updated 3 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 2 years ago
- [NeurIPS 2022] Compositional Generalization in Unsupervised Compositional Representation Learning: A Study on Disentanglement and Emergen…☆13Oct 7, 2022Updated 3 years ago
- Official PyTorch implementation of GENESIS and GENESIS-V2☆109Apr 13, 2022Updated 3 years ago
- [ICLR 2023 - UNOFFICIAL] Bridging the Gap to Real-World Object-Centric Learning☆23May 10, 2024Updated last year
- An implementation of several unsupervised object discovery models (Slot Attention, SLATE, GNM) in PyTorch with pre-trained models.☆14May 26, 2025Updated 9 months ago
- 🔥Benchmarking Unsupervised Obj Seg (NeurIPS 2022 & IJCV 2024)☆36Aug 26, 2025Updated 6 months ago
- ` google-research / slot-attention-video ` but in pytorch.☆18Aug 10, 2022Updated 3 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆47Mar 24, 2023Updated 2 years ago
- Code for the benchmark containing dataset, models and metrics for productive concept learning -- a kind of compositional reasoning task t…☆17Jul 22, 2021Updated 4 years ago
- [NeurIPS 2023] Self-supervised Object-Centric Learning for Videos☆32Nov 28, 2024Updated last year
- An implementation of the MONet model for unsupervised scene decomposition in PyTorch☆59May 16, 2022Updated 3 years ago
- EfficientMORL (ICML'21)☆22Nov 3, 2021Updated 4 years ago
- [NeurIPS 2020 (Spotlight)] PyTorch implementation of MulMON☆18Jan 26, 2025Updated last year
- Personal Python toolbox☆17Feb 2, 2026Updated last month
- Though language model text embeddings have revolutionized NLP research, their ability to capture high-level semantic information, such as…☆28Apr 28, 2023Updated 2 years ago
- Code for ICML2021 paper 'Commutative Lie Group VAE for Disentanglement Learning'.☆23Nov 2, 2022Updated 3 years ago
- This repository generates multi-view observations for static/dynamic CLEVR scenes.☆12Apr 17, 2023Updated 2 years ago
- Code for Static and Dynamic Concepts for Self-supervised Video Representation Learning.☆11Jul 28, 2022Updated 3 years ago
- [NeurIPS 2020] Better Set Representations For Relational Reasoning (https://arxiv.org/abs/2003.04448)☆31Jan 31, 2021Updated 5 years ago
- [NeurIPS 2022] code for "Visual Concepts Tokenization"☆23Oct 10, 2022Updated 3 years ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Sep 3, 2022Updated 3 years ago
- PyTorch implementation of Refine and Represent: Region-to-Object Representation Learning.☆21Jun 19, 2025Updated 9 months ago
- Entity Abstraction in Visual Model-Based Reinforcement Learning☆57Jan 1, 2021Updated 5 years ago
- ☆36Dec 26, 2022Updated 3 years ago
- Neural Networks that convert input movies into Physical Scene Graphs (PSGs)☆63Feb 26, 2021Updated 5 years ago
- Solving Physics Puzzles by Reasoning about Paths (NeurIPS 2020 workshop)☆14Jun 28, 2022Updated 3 years ago