Zhaowei Cai

semi-vit. PyTorch implementation of Semi-supervised Vision Transformers

omni-detr. PyTorch implementation of Omni-DETR for omni-supervised object detection: https://arxiv.org/abs/2203.16089

exponential-moving-average-normalization. PyTorch implementation of EMAN for self-supervised and semi-supervised learning: https://arxiv.org/abs/2101.08482

xmodaler. X-modaler is a versatile and high-performance codebase for cross-modal analytics(e.g., image captioning, video captioning, vision-language pre-training, visual question answering, visual commonsense reasoning, and cross-modal retrieval).

mmf. A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)

fairseq. Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

transformers. 🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

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

mmselfsup. OpenMMLab Self-Supervised Learning Toolbox and Benchmark

cnn-explainer. Learning Convolutional Neural Networks with Interactive Visualization.

moco. PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

detectron2. Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

pytorch_geometric. Graph Neural Network Library for PyTorch

video-nonlocal-net. Non-local Neural Networks for Video Classification

maskrcnn-benchmark. Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

Detectron. FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.

mmdetection. OpenMMLab Detection Toolbox and Benchmark

tensorpack. A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility