lucidrains / slot-attention
Implementation of Slot Attention from GoogleAI
☆394Updated 2 months ago
Related projects ⓘ
Alternatives and complementary repositories for slot-attention
- EsViT: Efficient self-supervised Vision Transformers☆408Updated last year
- Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)☆449Updated 2 years ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆252Updated 3 years ago
- Pytorch Implementation of paper "Object-Centric Learning with Slot Attention"☆82Updated last year
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆548Updated last year
- Repository for "Space-Time Correspondence as a Contrastive Random Walk" (NeurIPS 2020)☆268Updated 2 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,092Updated last year
- A Pytorch-Lightning implementation of self-supervised algorithms☆535Updated 2 years ago
- Code to reproduce the results in the FAIR research papers "Semi-Supervised Learning of Visual Features by Non-Parametrically Predicting V…☆487Updated last year
- ☆311Updated 7 months ago
- ☆152Updated last year
- Understanding Training Dynamics of Deep ReLU Networks☆278Updated 2 weeks ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆203Updated last year
- Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"☆417Updated 2 years ago
- Dense Contrastive Learning (DenseCL) for self-supervised representation learning, CVPR 2021 Oral.☆546Updated 10 months ago
- Pytorch implementation of LOST unsupervised object discovery method☆236Updated last year
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆806Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆404Updated last month
- Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)☆336Updated 2 months ago
- Self-supervised vIsion Transformer (SiT)☆324Updated last year
- VQVAEs, GumbelSoftmaxes and friends☆533Updated 2 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…☆191Updated 3 years ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆480Updated 2 years ago
- PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results☆191Updated 6 months ago
- Self-labelling via simultaneous clustering and representation learning. (ICLR 2020)☆532Updated 3 months ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆437Updated 10 months ago
- FiLM: Visual Reasoning with a General Conditioning Layer☆325Updated 2 years ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆257Updated 2 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 …☆82Updated last year
- [CVPR 2021] VirTex: Learning Visual Representations from Textual Annotations☆557Updated 10 months ago