Dazhi's starred repositories
graph-based-nn
Graph Convolutional Networks (GCNs)
pytorch-image-models
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
EfficientDet-Pytorch
EfficientDet implementation for object detection in Pytorch. EfficientNet as backbone. BiFPN.
detectron2_backbone
detectron2 backbone: resnet18, efficientnet, hrnet, mobilenet v2, resnest, bifpn
maskrcnn-benchmark
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
Student-Class-Behavior-Dataset
Coming soon
lmdb-embeddings
Fast word vectors with little memory usage in Python
ABSA-PyTorch
Aspect Based Sentiment Analysis, PyTorch Implementations. 基于方面的情感分析，使用PyTorch实现。
External-Attention-pytorch
🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐
attention-transfer
Improving Convolutional Networks via Attention Transfer (ICLR 2017)
AttentioNN
All about attention in neural networks. Soft attention, attention maps, local and global attention and multi-head attention.
face-alignment
:fire: 2D and 3D Face alignment library build using pytorch
HRNet-Image-Classification
Train the HRNet model on ImageNet
PoseEstimation-FCN-Pytorch
Resnet-34 with light-weighted decoder for pose estimation. (Pytorch)
Angular-Penalty-Softmax-Losses-Pytorch
Angular penalty loss functions in Pytorch (ArcFace, SphereFace, Additive Margin, CosFace)
classification_models
Classification models trained on ImageNet. Keras.
VGGFace2-pytorch
PyTorch Face Recognizer based on 'VGGFace2: A dataset for recognising faces across pose and age'
pretrained-models.pytorch
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
senet.pytorch
PyTorch implementation of SENet
detectron2
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
bottom-up-attention.pytorch
A PyTorch reimplementation of bottom-up-attention models
bottom-up-attention
Bottom-up attention model for image captioning and VQA, based on Faster R-CNN and Visual Genome
ActionBERT
Transformer for Action Recognition in PyTorch
EfficientNet-PyTorch
A PyTorch implementation of EfficientNet
FaceDetection-DSFD
腾讯优图高精度双分支人脸检测器
Pytorch-RNN-text-classification
Word Embedding + LSTM + FC