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,550
qubvel-org/segmentation_models.pytorch A PyTorch library for building and training neural networks for image segmentation tasks. 9,696
lukemelas/efficientnet-pytorch A PyTorch implementation of EfficientNet convolutional neural networks 7,921
moskomule/senet.pytorch An implementation of Squeeze-and-Excitation Networks (SE-Nets) for deep learning image classification tasks 2,285
facebookresearch/dinov2 A PyTorch implementation of a self-supervised learning method for learning robust visual features without supervision. 9,211
jfzhang95/pytorch-deeplab-xception A PyTorch implementation of the DeepLab-V3-Plus model with support for multiple backbones and datasets 2,910
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,251
cszn/kair Image restoration toolbox with training and testing codes for various deep learning-based methods 2,957
meetps/pytorch-semseg Implementations of popular semantic segmentation architectures using PyTorch 3,393
pytorch/examples A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. 22,428
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 83,959
lim0606/torch-inception-resnet-v2 This project re-implements the Inception-ResNet-v2 model from scratch using PyTorch and trains it on ImageNet. 52
dbiir/uer-py A toolkit for building pre-training models and fine-tuning them on downstream tasks in natural language processing 3,001
ahkarami/deep-learning-in-production A collection of notes and references on deploying deep learning models in production environments 4,306
diego999/pygat An implementation of the Graph Attention Network model using PyTorch. 2,914