A deep learning code base, mainly for paper replication, in the areas of image recognition, object detection, image segmentation, self-supervision, etc. Each project can be run independently, and t…

Code for paper "DeepEMD: Few-Shot Image Classification with Differentiable Earth Mover's Distance and Structured Classifiers", CVPR2020

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

Code release for "BoxeR: Box-Attention for 2D and 3D Transformers"

论文复现，多机多卡

Many Class Activation Map methods implemented in Pytorch for CNNs and Vision Transformers. Including Grad-CAM, Grad-CAM++, Score-CAM, Ablation-CAM and XGrad-CAM

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

CV backbones including GhostNet, TinyNet and TNT, developed by Huawei Noah's Ark Lab.

Siamese and triplet networks with online pair/triplet mining in PyTorch

Dense Relation Distillation with Context-aware Aggregation for Few-Shot Object Detection, CVPR 2021

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

BossNAS: Exploring Hybrid CNN-transformers with Block-wisely Self-supervised Neural Architecture Search (ICCV 2021)

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

deep learning for image processing including classification and object-detection etc.

Official Pytorch Implementation of: "ImageNet-21K Pretraining for the Masses"(NeurIPS, 2021) paper

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

A PyTorch implementation of EfficientNet and EfficientNetV2 (coming soon!)

label-smooth, amsoftmax, focal-loss, triplet-loss, lovasz-softmax. Maybe useful

A simple pytorch implementation for multi-class focal loss

Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet…

[CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator

This is the official PyTorch implementation of the paper "TransFG: A Transformer Architecture for Fine-grained Recognition" (Ju He, Jie-Neng Chen, Shuai Liu, Adam Kortylewski, Cheng Yang, Yutong Ba…

A simplified implemention of Faster R-CNN that replicate performance from origin paper

GitHub repository for the Hyper-Kvasir dataset.

PyTorch implementation of SENet

Non-official implement of Paper：CBAM: Convolutional Block Attention Module