Awesome Open Source

Programming Languages

Search results for pytorch imagenet

333 search results found

Pytorch Image Models ⭐ 29,680

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Cvat ⭐ 11,169

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

Segmentation_models.pytorch ⭐ 8,226

Segmentation models with pretrained backbones. PyTorch.

Pretrained Models.pytorch ⭐ 8,094

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

Efficientnet Pytorch ⭐ 6,577

A PyTorch implementation of EfficientNet and EfficientNetV2 (coming soon!)

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

Super Gradients ⭐ 4,125

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.

Efficient Ai Backbones ⭐ 3,770

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

Official DeiT repository

Damo Yolo ⭐ 3,598

DAMO-YOLO: a fast and accurate object detection method with some new techs, including NAS backbones, efficient RepGFPN, ZeroHead, AlignedOTA, and distillation enhancement.

Pytorch Studiogan ⭐ 3,226

StudioGAN is a Pytorch library providing implementations of representative Generative Adversarial Networks (GANs) for conditional/unconditional image generation.

Darts ⭐ 2,775

Differentiable architecture search for convolutional and recurrent networks

Flops Counter.pytorch ⭐ 2,606

Flops counter for convolutional networks in pytorch framework

Imgclsmob ⭐ 2,399

Sandbox for training deep learning networks

Biggan Pytorch ⭐ 2,356

The author's officially unofficial PyTorch BigGAN implementation.

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

Ml Cvnets ⭐ 1,543

CVNets: A library for training computer vision networks

Senet.pytorch ⭐ 1,494

PyTorch implementation of SENet

Big_transfer ⭐ 1,421

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

Fast Autoaugment ⭐ 1,358

Official Implementation of 'Fast AutoAugment' in PyTorch.

Pytorch Classification ⭐ 1,244

Classification with PyTorch.

Data Efficient Gans ⭐ 1,231

[NeurIPS 2020] Differentiable Augmentation for Data-Efficient GAN Training

Torchdistill ⭐ 1,171

A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆22 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.

Attention Transfer ⭐ 1,120

Improving Convolutional Networks via Attention Transfer (ICLR 2017)

Vit Pytorch ⭐ 984

Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)

Pytorch_image_classification ⭐ 979

PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MN

Personalized federated learning simulation platform with non-IID and unbalanced dataset

PyTorch Implementation of "Lossless Image Compression through Super-Resolution"

Learningtocompare_fsl ⭐ 891

PyTorch code for CVPR 2018 paper: Learning to Compare: Relation Network for Few-Shot Learning (Few-Shot Learning part)

Pytorch Distributed ⭐ 859

A quickstart and benchmark for pytorch distributed training.

Addernet ⭐ 857

Code for paper " AdderNet: Do We Really Need Multiplications in Deep Learning?"

Pytorch2keras ⭐ 848

PyTorch to Keras model convertor

Efficientformer ⭐ 790

EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]

Dbnet.pytorch ⭐ 772

A pytorch re-implementation of Real-time Scene Text Detection with Differentiable Binarization

Switchable Normalization ⭐ 771

Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10779

Dl Engineer Guidebook ⭐ 770

深度学习工程师生存指南

Efficient Pytorch ⭐ 769

My best practice of training large dataset using PyTorch.

Robustness ⭐ 728

Corruption and Perturbation Robustness (ICLR 2019)

Neural Network Compression Framework for enhanced OpenVINO™ inference

Howtotrainyourmamlpytorch ⭐ 707

The original code for the paper "How to train your MAML" along with a replication of the original "Model Agnostic Meta Learning" (MAML) paper in Pytorch.

Classification_models ⭐ 706

Classification models trained on ImageNet. Keras.

Condensenet ⭐ 655

CondenseNet: Light weighted CNN for mobile devices

Mdistiller ⭐ 634

The official implementation of [CVPR2022] Decoupled Knowledge Distillation https://arxiv.org/abs/2203.08679 and [ICCV2023] DOT: A Distillation-Oriented Trainer https://openaccess.thecvf.com/content/ICCV2023/pap

Cbam.pytorch ⭐ 630

Non-official implement of Paper：CBAM: Convolutional Block Attention Module

Cutmix Pytorch ⭐ 611

Official Pytorch implementation of CutMix regularizer

Meal V2 ⭐ 603

MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks

Pytorch Cnn Finetune ⭐ 580

Fine-tune pretrained Convolutional Neural Networks with PyTorch

Functional Zoo ⭐ 557

PyTorch and Tensorflow functional model definitions

Pytorch Randaugment ⭐ 557

Unofficial PyTorch Reimplementation of RandAugment.

Gluoncv Torch ⭐ 548

PyTorch API for GluonCV Models

Openmixup ⭐ 538

CAIRI Supervised, Semi- and Self-Supervised Visual Representation Learning Toolbox and Benchmark

Filter Pruning Geometric Median ⭐ 489

Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Acceleration (CVPR 2019 Oral)

Pytorch Mobilenet V3 ⭐ 481

MobileNetV3 in pytorch and ImageNet pretrained models

Shufflenet_v2_pytorch_caffe ⭐ 480

ShuffleNet-V2 for both PyTorch and Caffe.

Lemniscate.pytorch ⭐ 467

Unsupervised Feature Learning via Non-parametric Instance Discrimination

Neural Backed Decision Trees ⭐ 445

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

Osvos Pytorch ⭐ 434

PyTorch implementation of One-Shot Video Object Segmentation (OSVOS)

Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models

Resnext.pytorch ⭐ 388

Reproduces ResNet-V3 with pytorch

Residualmaskingnetwork ⭐ 387

ICPR 2020: Facial Expression Recognition using Residual Masking Network

Libtorchsegmentation ⭐ 380

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.

Pytorch Pretrained Vit ⭐ 378

Vision Transformer (ViT) in PyTorch

Espnetv2 ⭐ 375

A light-weight, power efficient, and general purpose convolutional neural network

Shufflenet V2 Pytorch ⭐ 369

Perfect implement. Model shared. x0.5 (Top1:60.646) and 1.0x (Top1:69.402).

Mobilenetv2.pytorch ⭐ 352

72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models

Bag_of_tricks_for_image_classification_with_convolutional_neural_networks ⭐ 352

experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN acc

Deephash Pytorch ⭐ 348

Implementation of Some Deep Hash Algorithms, Including DPSH、DSH、DHN、HashNet、DSDH、DTSH、DFH、GreedyHash、CSQ.

Benchmarking Keras Pytorch ⭐ 342

🔥 Reproducibly benchmarking Keras and PyTorch models

Tokenlabeling ⭐ 342

Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"

Pytorch I Revnet ⭐ 338

i-RevNet Pytorch Code

Sparse_learning ⭐ 334

Sparse learning library and sparse momentum resources.

Xnor Net Pytorch ⭐ 331

PyTorch Implementation of XNOR-Net

Densenet Pytorch ⭐ 306

A PyTorch Implementation for Densely Connected Convolutional Networks (DenseNets)

Pytorch Hardnet ⭐ 305

35% faster than ResNet: Harmonic DenseNet, A low memory traffic network

Pretorched X ⭐ 299

Pretrained Image & Video ConvNets and GANs for PyTorch: NASNet, ResNeXt (2D + 3D), ResNet (2D + 3D), InceptionV4, InceptionResnetV2, Xception, DPN, NonLocalNets, R(2+1)D nets, MultiView CNNs, Temporal Relation Networks, BigGANs StyleGANs, etc.

Pnasnet.pytorch ⭐ 299

PyTorch implementation of PNASNet-5 on ImageNet

Convnet.pytorch ⭐ 279

ConvNet training using pytorch

Efficientnets Pytorch ⭐ 275

A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."

Randwirenn ⭐ 259

Pytorch Implementation of: "Exploring Randomly Wired Neural Networks for Image Recognition"

Styletransfertrilogy ⭐ 254

风格迁移三部曲

Cgnl Network.pytorch ⭐ 253

Compact Generalized Non-local Network (NIPS 2018)

Shufflenet ⭐ 251

ShuffleNet in PyTorch. Based on https://arxiv.org/abs/1707.01083

Attention_branch_network ⭐ 236

Attention Branch Network (CIFAR100, ImageNet models)

Soft Filter Pruning ⭐ 235

Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks

Senet Pytorch ⭐ 232

Parametric Contrastive Learning ⭐ 220

Parametric Contrastive Learning (ICCV2021) & GPaCo (TPAMI 2023)

Pyramidnet Pytorch ⭐ 215

A PyTorch implementation for PyramidNets (Deep Pyramidal Residual Networks, https://arxiv.org/abs/1610.02915)

Learningtocompare Pytorch ⭐ 214

Pytorch Implementation for CVPR2018 Paper: Learning to Compare: Relation Network for Few-Shot Learning

MoGA: Searching Beyond MobileNetV3

Pyramidal Convolution: Rethinking Convolutional Neural Networks for Visual Recognition (https://arxiv.org/pdf/2006.11538.pdf)

Sequential Imagenet Dataloader ⭐ 199

A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.

Mobilenetv3 Pytorch ⭐ 197

Implementing Searching for MobileNetV3 paper using Pytorch

Octconv.pytorch ⭐ 187

PyTorch implementation of Octave Convolution with pre-trained Oct-ResNet and Oct-MobileNet models

Code and pretrained models for LIP: Local Importance-based Pooling (ICCV 19)

Pytorch Dpn Pretrained ⭐ 182

Dual Path Networks (DPN) supporting pretrained weights converted from original MXNet implementation

Stand Alone Self Attention ⭐ 171

Implementing Stand-Alone Self-Attention in Vision Models using Pytorch

Taylor_pruning ⭐ 169

Pruning Neural Networks with Taylor criterion in Pytorch

Selecsls Pytorch ⭐ 165

Reference ImageNet implementation of SelecSLS CNN architecture proposed in the SIGGRAPH 2020 paper "XNect: Real-time Multi-Person 3D Motion Capture with a Single RGB Camera". The repository also includes code for pruning the model based on implicit sparsity emerging from adaptive gradient descent methods, as detailed in the CVPR 2019 paper "On implicit filter level sparsity in Convolutional Neural Networks".

Vision Transformer Pytorch ⭐ 164

Pytorch version of Vision Transformer (ViT) with pretrained models. This is part of CASL (https://casl-project.github.io/) and ASYML project.

Shufflenetv2 Pytorch ⭐ 162

Implementation of ShuffleNetV2 for pytorch

Related Searches

Python Pytorch (15,943)

Deep Learning Pytorch (7,533)

Jupyter Notebook Pytorch (4,892)

Machine Learning Pytorch (2,934)

Dataset Pytorch (1,848)

Pytorch Convolutional Neural Networks (1,777)

Pytorch Neural Network (1,631)

Pytorch Natural Language Processing (1,408)

Python Imagenet (1,293)

Pytorch Computer Vision (1,230)

1-100 of 333 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.