pretrained-weights

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

Segmentation models with pretrained backbones. PyTorch.

Classification models trained on ImageNet. Keras.

Easily train or fine-tune SOTA computer vision models with one open source training library

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

YOLOv3 implementation in TensorFlow 2.3.1

Pretrained TorchVision models on CIFAR10 dataset (with weights)

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

Code for "Real-time self-adaptive deep stereo" - CVPR 2019 (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.

Hierarchical Image Pyramid Transformer - CVPR 2022 (Oral)

Top 1 solution of Chexpert

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

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

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.

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

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.

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

Multi-label classification based on timm.

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