pretrained-weights topic

List pretrained-weights repositories
trafficstars

segmentation_models.pytorch

9.5k
Stars
1.7k
Forks
Watchers

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

HIPT

466
Stars
83
Forks
Watchers

Hierarchical Image Pyramid Transformer - CVPR 2022 (Oral)

classification_models

1.2k
Stars
305
Forks
Watchers

Classification models trained on ImageNet. Keras.

pytorch-image-models

31.9k
Stars
4.7k
Forks
Watchers

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT)...

Real-time-self-adaptive-deep-stereo

417
Stars
75
Forks
Watchers

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

LibtorchSegmentation

403
Stars
85
Forks
Watchers

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.

neural-backed-decision-trees

602
Stars
131
Forks
Watchers

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

PyTorch_CIFAR10

610
Stars
153
Forks
Watchers

Pretrained TorchVision models on CIFAR10 dataset (with weights)

TensorFlow-2.x-YOLOv3

593
Stars
326
Forks
Watchers

YOLOv3 implementation in TensorFlow 2.3.1

Entity

675
Stars
56
Forks
Watchers

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