zhfeing

Bootstrapping-ViTs-pytorch

Official pytorch implementation for CVPR2022 paper "Bootstrapping ViTs: Towards Liberating Vision Transformers from Pre-training"

SchemaNet-PyTorch

Official PyTorch implementation of paper "Schema Inference for Interpretable Image Classification" (ICLR 2023)

cv-lib-PyTorch

Utils for computer vision in deep learning training and evaluating, e.g. classification, object detection...

data-visualization

Visualization of sci-data

DNN-inspector

linux-windows-notes

notes and tools on linux and windows

Transformer-KA-PyTorch

Official PyTorch implementation of paper "Knowledge Amalgamation for Object Detection with Transformers" (TIP)

distill-engine

engine for knowledge distillation

pysmg

Pytorch Semantic Segmentation Library

pytorch-distributed-training

Cross-Compiling-Qt-for-Raspberry-Pi-4

A guide to cross-compile Qt 5.15.0 for the Raspberry Pi 4 Model B

cv-project-segmentation

DeepLearningExamples

Deep Learning Examples

Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.

detectron2

Detectron2 is FAIR's next-generation platform for object detection and segmentation.

detr

End-to-End Object Detection with Transformers

External-Attention-pytorch

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

flash-attention

Fast and memory-efficient exact attention

how-do-vits-work

(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"

loss-landscape

Code for visualizing the loss landscape of neural nets

nanoGPT

The simplest, fastest repository for training/finetuning medium-sized GPTs.

pytorch-cifar100

Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet, NasNet, Residual Attention Network, SENet)

pytorch-classification

Classification with PyTorch.

RepDistiller

[ICLR 2020] Contrastive Representation Distillation (CRD), and benchmark of recent knowledge distillation methods

segmentation_models.pytorch

Segmentation models with pretrained backbones. PyTorch.

ssd.pytorch

A PyTorch Implementation of Single Shot MultiBox Detector

TransferbilityFromAttributionMaps

(NeurIPS 2019) Deep Model Transferbility from Attribution Maps

unified-io-inference

up-detr

[CVPR2021 Oral] UP-DETR: Unsupervised Pre-training for Object Detection with Transformers

zhfeing.github.io