Dream1230v

NAM

bottleneck-transformer-pytorch

Implementation of Bottleneck Transformer in Pytorch

how_attentive_are_gats

Code for the paper "How Attentive are Graph Attention Networks?" (ICLR'2022)

pytorch_geometric

Graph Neural Network Library for PyTorch

Non-local_pytorch

Implementation of Non-local Block.

External-Attention-pytorch

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

gluon-cv

Gluon CV Toolkit

mixup

Implementation of the mixup training method

SimAM

The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Networks".

FasterNet

[CVPR 2023] Code for PConv and FasterNet

cap

API-Net

Learning Attentive Pairwise Interaction for Fine-Grained Classification, AAAI-2020

PyTorch-Networks

Pytorch implementation of cnn network

SMPConv

[CVPR2023] "SMPConv: Self-moving Point Representations for Continuous Convolution"

open-aff

code and trained models for "Attentional Feature Fusion"

alias_free_convnets

iresnet

Improved Residual Networks (https://arxiv.org/pdf/2004.04989.pdf)

slimmable_networks

Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019

mage

A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis

MAE-pytorch

Unofficial PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners

SimMIM

This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".

mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

BiFormer

[CVPR 2023] Official code release of our paper "BiFormer: Vision Transformer with Bi-Level Routing Attention"

dinov2

PyTorch code and models for the DINOv2 self-supervised learning method.

imgaug

Image augmentation for machine learning experiments.

ACR

Pytorch implementation for "Towards Realistic Long-Tailed Semi-Supervised Learning: Consistency Is All You Need" (CVPR 2023)

DTN

[AAAI-2020, Oral] Diversity Transfer Network for Few-Shot Learning

ConvNeXt-V2

Code release for ConvNeXt V2 model

dilateformer