lucidrains / slot-attentionLinks
Implementation of Slot Attention from GoogleAI
☆476Updated last year
Alternatives and similar repositories for slot-attention
Users that are interested in slot-attention are comparing it to the libraries listed below
Sorting:
- Pytorch Implementation of paper "Object-Centric Learning with Slot Attention"☆106Updated 2 years ago
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆610Updated 3 years ago
- FiLM: Visual Reasoning with a General Conditioning Layer☆418Updated 4 years ago
- ☆178Updated 2 years ago
- Simple project webpage template. Originally used in Colorful Image Colorization. ECCV, 2016.☆486Updated 5 years ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆504Updated 2 years ago
- EsViT: Efficient self-supervised Vision Transformers☆411Updated 2 years ago
- Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)☆463Updated 3 years ago
- Repository for "Space-Time Correspondence as a Contrastive Random Walk" (NeurIPS 2020)☆273Updated 4 years ago
- Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"☆427Updated 3 years ago
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆195Updated 4 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,191Updated 2 years ago
- Understanding Training Dynamics of Deep ReLU Networks☆306Updated 2 months ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆211Updated last year
- 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 …☆88Updated 3 years ago
- Supplementary code for the paper "Hyperbolic Image Embeddings".☆390Updated 4 years ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆501Updated 3 years ago
- ☆340Updated 5 months ago
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆358Updated 8 months ago
- PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference☆59Updated 3 years ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆279Updated 4 years ago
- Library for the training and evaluation of object-centric models (ICML 2022)☆71Updated 2 years ago
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆614Updated last year
- A Pytorch-Lightning implementation of self-supervised algorithms☆545Updated 3 years ago
- A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.☆260Updated 5 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆210Updated 2 years ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆237Updated 4 years ago
- CATER: A diagnostic dataset for Compositional Actions and TEmporal Reasoning☆108Updated 5 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆445Updated last year
- Code and model definition for converting the official BYOL weights to PyTorch☆33Updated 4 years ago