Neural networks implementations in PyTorch

Backbones:

• ResNet
• MobileNetV2
• MobileNetV3
• Visual Transformer (ViT)
• Swin Transformer: Hierarchical Vision Transformer using Shifted Windows

Object Detection

• Objects as Points
• DETR: End-to-End Object Detection with Transformers

Semantic Segmentations

• SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformers
• DeepLabv3+

NLP:

• NLP Transformer (Attention is all you need)