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

OpenMMLab's next-generation platform for general 3D object detection.

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

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Object Detection and Instance Segmentation.

[CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)

OneNet: End-to-End One-Stage Object Detection

Self-Supervised Learning Toolbox and Benchmark

Multiple-Object Tracking with Transformer

End-to-End Object Detection with Transformers

Support PointRend, Fast_SCNN, HRNet, Deeplabv3_plus(xception, resnet, mobilenet), ContextNet, FPENet, DABNet, EdaNet, ENet, Espnetv2, RefineNet, UNet, DANet, HRNet, DFANet, HardNet, LedNet, OCNet, …

Tightness-aware Evaluation Protocol for Scene Text Detection (CVPR 2019)

Classification with PyTorch.

Semantic Segmentation ported from PyTorchCV

Deep learning library featuring a higher-level API for TensorFlow.