lucidrains / deformable-attentionView external linksLinks
Implementation of Deformable Attention in Pytorch from the paper "Vision Transformer with Deformable Attention"
☆372Feb 3, 2025Updated last year
Alternatives and similar repositories for deformable-attention
Users that are interested in deformable-attention are comparing it to the libraries listed below
Sorting:
- Repository of Vision Transformer with Deformable Attention (CVPR2022) and DAT++: Spatially Dynamic Vision Transformerwith Deformable Atte…☆925Apr 17, 2024Updated last year
- Deformable DETR: Deformable Transformers for End-to-End Object Detection.☆3,892May 16, 2024Updated last year
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆36Sep 27, 2021Updated 4 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆51May 10, 2022Updated 3 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Apr 6, 2022Updated 3 years ago
- Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification tasks, de…☆102Apr 22, 2022Updated 3 years ago
- [ICME 2022] code for the paper, SimVit: Exploring a simple vision transformer with sliding windows.☆68Oct 11, 2022Updated 3 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆57Jan 5, 2023Updated 3 years ago
- Local Attention - Flax module for Jax☆22May 26, 2021Updated 4 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆292May 3, 2024Updated last year
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆101Aug 1, 2023Updated 2 years ago
- Official PyTorch implementation of HCCNet: Efficient Semantic Matching with Hypercolumn Correlation (WACV '24 Oral, Best paper finalist (…☆11Apr 29, 2024Updated last year
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆220Feb 13, 2023Updated 3 years ago
- Implementation of Multistream Transformers in Pytorch☆54Jul 31, 2021Updated 4 years ago
- Implementation of Differentiable Sign-Distance Function Rendering - in Pytorch☆70May 9, 2022Updated 3 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,175May 15, 2024Updated last year
- [ICCV 2023] Official implementation of the paper "DFA3D: 3D Deformable Attention For 2D-to-3D Feature Lifting"☆179Apr 12, 2025Updated 10 months ago
- [ICCV 2023] The official repository of our paper "UMC: A Unified Bandwidth-efficient and Multi-resolution based Collaborative Perception …☆14Aug 19, 2023Updated 2 years ago
- Implementation of Segformer, Attention + MLP neural network for segmentation, in Pytorch☆373Dec 9, 2022Updated 3 years ago
- Temporally Efficient Vision Transformer for Video Instance Segmentation, CVPR 2022, Oral☆239Mar 4, 2023Updated 2 years ago
- Implementation of RQ Transformer, proposed in the paper "Autoregressive Image Generation using Residual Quantization"☆124Apr 19, 2022Updated 3 years ago
- ☆31Oct 15, 2025Updated 4 months ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆98Dec 31, 2021Updated 4 years ago
- A method to increase the speed and lower the memory footprint of existing vision transformers.☆1,170Jun 17, 2024Updated last year
- This is the official repo of Panoptic SegFormer [CVPR'22]☆240Mar 3, 2022Updated 3 years ago
- [CVPR 2024] Deformable Convolution v4☆706May 17, 2024Updated last year
- PyTorch implementation of "Reference-based Image Super-Resolution with Deformable Attention Transformer (ECCV2022)"☆165Mar 28, 2024Updated last year
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆16Aug 3, 2021Updated 4 years ago
- Efficient Space-time Super Resolution using Flow and Mask Upsampling☆10Aug 29, 2021Updated 4 years ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆824Jul 14, 2022Updated 3 years ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆641Dec 19, 2025Updated last month
- [ICCV 2025] TurboTrain: Towards Efficient and Balanced Multi-Task Learning for Multi-Agent Perception and Prediction.☆16Jan 31, 2026Updated 2 weeks ago
- This branch of Asteroid contains code for the vocal harmony and chamber ensemble separation related papers.☆12Nov 7, 2024Updated last year
- PyTorch Implementation of Sparse DETR☆175Jan 3, 2024Updated 2 years ago
- Vector (and Scalar) Quantization, in Pytorch☆3,870Updated this week
- Implementation of Video Diffusion Models, Jonathan Ho's new paper extending DDPMs to Video Generation - in Pytorch☆1,376May 3, 2024Updated last year
- Code release for ConvNeXt model☆6,292Jan 8, 2023Updated 3 years ago
- Implementation of the Llama architecture with RLHF + Q-learning☆170Feb 1, 2025Updated last year
- Per-Pixel Classification is Not All You Need for Semantic Segmentation (NeurIPS 2021, spotlight)☆1,450Mar 11, 2022Updated 3 years ago