pretrained-models.pytorch

PyTorch Models

Provides pre-trained PyTorch models for various computer vision tasks and datasets.

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

GitHub

9k stars
214 watching
2k forks
Language: Python
last commit: over 2 years ago
imagenetinceptionpretrainedpytorchresnetresnext

Related projects:

Repository Description Stars
timesler/facenet-pytorch This project provides pre-trained PyTorch models for face detection and recognition tasks. 4,581
qubvel-org/segmentation_models.pytorch A comprehensive library for training and applying deep learning models for image segmentation 9,759
lukemelas/efficientnet-pytorch A PyTorch implementation of EfficientNet convolutional neural networks 7,942
moskomule/senet.pytorch An implementation of Squeeze-and-Excitation Networks (SE-Nets) for deep learning image classification tasks 2,290
facebookresearch/dinov2 A PyTorch implementation of a self-supervised learning method for learning robust visual features without supervision. 9,326
jfzhang95/pytorch-deeplab-xception A PyTorch implementation of the DeepLab-V3-Plus model with support for multiple backbones and datasets 2,917
yunjey/pytorch-tutorial A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. 30,336
cszn/kair Image restoration toolbox with training and testing codes for various deep learning-based methods 2,978
meetps/pytorch-semseg Implementations of popular semantic segmentation architectures using PyTorch 3,392
pytorch/examples A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. 22,490
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 84,572
lim0606/torch-inception-resnet-v2 Re-training Inception-ResNet-V2 model from scratch using PyTorch on limited computational resources 52
dbiir/uer-py A toolkit for building pre-training models and fine-tuning them on downstream tasks in natural language processing 3,010
ahkarami/deep-learning-in-production A collection of notes and references on deploying deep learning models in production environments 4,311
diego999/pygat An implementation of the Graph Attention Network model using PyTorch. 2,926