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 4 years ago
imagenetinceptionpretrainedpytorchresnetresnext

Related projects:

RepositoryDescriptionStars
timesler/facenet-pytorchThis project provides pre-trained PyTorch models for face detection and recognition tasks.4,611
qubvel-org/segmentation_models.pytorchA comprehensive library for training and applying deep learning models for image segmentation9,829
lukemelas/efficientnet-pytorchA PyTorch implementation of EfficientNet convolutional neural networks7,971
moskomule/senet.pytorchAn implementation of Squeeze-and-Excitation Networks (SE-Nets) for deep learning image classification tasks2,291
facebookresearch/dinov2A PyTorch implementation of a self-supervised learning method for learning robust visual features without supervision.9,425
jfzhang95/pytorch-deeplab-xceptionA PyTorch implementation of the DeepLab-V3-Plus model with support for multiple backbones and datasets2,919
yunjey/pytorch-tutorialA comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage.30,401
cszn/kairImage restoration toolbox with training and testing codes for various deep learning-based methods2,994
meetps/pytorch-semsegImplementations of popular semantic segmentation architectures using PyTorch3,400
pytorch/examplesA collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning.22,530
pytorch/pytorchA Python library providing tensors and dynamic neural networks with strong GPU acceleration84,978
lim0606/torch-inception-resnet-v2A re-trained version of the Inception-ResNet-V2 model using Torch, with notes on training conditions and experimental results.52
dbiir/uer-pyA toolkit for building pre-training models and fine-tuning them on downstream tasks in natural language processing3,018
ahkarami/deep-learning-in-productionA collection of notes and references on deploying deep learning models in production environments4,313
diego999/pygatAn implementation of the Graph Attention Network model using PyTorch.2,935