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

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

Classification models trained on ImageNet. Keras.

EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation

Pretrained TorchVision models on CIFAR10 dataset (with weights)

YOLOv3 implementation in TensorFlow 2.3.1

Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet

Hierarchical Image Pyramid Transformer - CVPR 2022 (Oral)

A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.

Code for "Real-time self-adaptive deep stereo" - CVPR 2019 (ORAL)

PyTorch Lightning Optical Flow models, scripts, and pretrained weights.

An Out-of-the-Box Replication of GANimation using PyTorch, pretrained weights are available!

Top 1 solution of Chexpert

A Beautiful Flask Web API for Yolov7 (and custom) models

https://towardsdatascience.com/tutorial-build-an-object-detection-system-using-yolo-9a930513643a

Self-Supervised Vision Transformers Learn Visual Concepts in Histopathology (LMRL Workshop, NeurIPS 2021)

A native Tensorflow implementation of semantic segmentation according to Multi-Scale Context Aggregation by Dilated Convolutions (2016). Optionally uses the pretrained weights by the authors.

Pretrained is the most complete and frequently updated list of pretrained top-performing models. Tensorflow, Theano and others. Want to add your model? File an issue, and we will add it.

Multi-label classification based on timm.

C++ trainable detection library based on libtorch (or pytorch c++). Yolov4 tiny provided now.