ivan-chai / torch-linear-assignment
Batch computation of the linear assignment problem on GPU.
☆42Updated last week
Related projects ⓘ
Alternatives and complementary repositories for torch-linear-assignment
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆213Updated 3 weeks ago
- [ICLR 2023 Spotlight] GPViT: A High Resolution Non-Hierarchical Vision Transformer with Group Propagation☆97Updated last year
- Demystify RAM Usage in Multi-Process Data Loaders☆180Updated last year
- A compilation of network architectures for vision and others without usage of self-attention mechanism☆77Updated last year
- A simple minimal implementation of Reversible Vision Transformers☆117Updated 8 months ago
- Code and models for the paper "The effectiveness of MAE pre-pretraining for billion-scale pretraining" https://arxiv.org/abs/2303.13496☆81Updated 3 months ago
- This repo contains the official implementation of ICLR 2024 paper "Is ImageNet worth 1 video? Learning strong image encoders from 1 long …☆61Updated 6 months ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆136Updated 2 years ago
- (ICLR 2023) Official PyTorch implementation of "What Do Self-Supervised Vision Transformers Learn?"☆102Updated 8 months ago
- [ICML 2024] This repository includes the official implementation of our paper "Rejuvenating image-GPT as Strong Visual Representation Lea…☆98Updated 6 months ago
- A PyTorch implementation of Mugs proposed by our paper "Mugs: A Multi-Granular Self-Supervised Learning Framework".☆83Updated 9 months ago
- PyTorch implementation of R-MAE https//arxiv.org/abs/2306.05411☆109Updated last year
- An official code release of the paper RGB no more: Minimally Decoded JPEG Vision Transformers☆49Updated last year
- ☆49Updated last year
- MixMIM: Mixed and Masked Image Modeling for Efficient Visual Representation Learning☆129Updated last year
- [CVPR2022 - Oral] Official Jax Implementation of Learned Queries for Efficient Local Attention☆114Updated 2 years ago
- [CVPR24] Official Implementation of GEM (Grounding Everything Module)☆86Updated 3 weeks ago
- The official repo for [TPAMI'23] "Vision Transformer with Quadrangle Attention"☆177Updated 7 months ago
- Transformers w/o Attention, based fully on MLPs☆89Updated 7 months ago
- a PyTorch re-implementation of ECCV 2022 paper based on Detectron2: k-means mask Transformer.☆67Updated last year
- ☆57Updated 2 years ago
- Visualizing representations with diffusion based conditional generative model.☆84Updated last year
- code release of research paper "Exploring Long-Sequence Masked Autoencoders"☆99Updated 2 years ago
- Official repository of paper "Subobject-level Image Tokenization"☆62Updated 6 months ago
- [CVPR 2023] This repository includes the official implementation our paper "Masked Autoencoders Enable Efficient Knowledge Distillers"☆99Updated last year
- A Contrastive Learning Boost from Intermediate Pre-Trained Representations☆39Updated 2 months ago
- Time Does Tell: Self-Supervised Time-Tuning of Dense Image Representations ICCV23☆26Updated last month
- Open-source code for Generic Grouping Network (GGN, CVPR 2022)☆109Updated 7 months ago
- Pytorch implementation of LOST unsupervised object discovery method☆237Updated last year
- Code for the paper "Hyperbolic Image-Text Representations", Desai et al, ICML 2023☆135Updated last year