Awesome Open Source

Programming Languages

Search results for pytorch image classification

image-classification x

242 search results found

Pytorch Image Models ⭐ 29,490

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

Ultralytics ⭐ 18,719

NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

Vit Pytorch ⭐ 16,298

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Pytorch Book ⭐ 11,409

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)

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.

Pytorch Grad Cam ⭐ 8,723

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

Techniques ⭐ 7,678

Techniques for deep learning with satellite & aerial imagery

Gluon Cv ⭐ 5,422

Gluon CV Toolkit

Super Gradients ⭐ 4,125

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

Pytorch Cifar100 ⭐ 3,687

Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet, NasNet, Residual Attention Network, SENet, WideResNet)

Notebooks ⭐ 3,677

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models like Grounding DINO and SAM.

Catalyst ⭐ 3,151

Accelerated deep learning R&D

Mmpretrain ⭐ 3,114

OpenMMLab Pre-training Toolbox and Benchmark

Darts ⭐ 2,775

Differentiable architecture search for convolutional and recurrent networks

Deepdetect ⭐ 2,492

Deep Learning API and Server in C++14 support for Caffe, PyTorch,TensorRT, Dlib, NCNN, Tensorflow, XGBoost and TSNE

Imgclsmob ⭐ 2,399

Sandbox for training deep learning networks

Awesome Image Classification ⭐ 2,138

A curated list of deep learning image classification papers and codes

Imbalanced Dataset Sampler ⭐ 2,073

A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.

Sparseml ⭐ 1,910

Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models

Ailia Models ⭐ 1,708

The collection of pre-trained, state-of-the-art AI models for ailia SDK

Pytorch Toolbelt ⭐ 1,458

PyTorch extensions for fast R&D prototyping and Kaggle farming

Fast Autoaugment ⭐ 1,358

Official Implementation of 'Fast AutoAugment' in PyTorch.

Autodistill ⭐ 1,286

Images to inference with no labeling (use foundation models to train supervised models)

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.

Training_extensions ⭐ 1,119

Train, Evaluate, Optimize, Deploy Computer Vision Models via OpenVINO™

Involution ⭐ 1,106

[CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator

Awesome Backbones ⭐ 1,079

Integrate deep learning models for image classification | Backbone learning/comparison/magic modification project

Pytorch_classification ⭐ 1,055

利用pytorch实现图像分类的一个完整的代码，训练，预测，TTA，模型融合，模型部署，cnn提取特

Cv Pretrained Model ⭐ 991

A collection of computer vision pre-trained models.

Poolformer ⭐ 981

PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)

Quickdraw ⭐ 851

Implementation of Quickdraw - an online game developed by Google

Libfewshot ⭐ 771

LibFewShot: A Comprehensive Library for Few-shot Learning. TPAMI 2023.

Torchxrayvision ⭐ 760

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

Easy Few Shot Learning ⭐ 737

Ready-to-use code and tutorial notebooks to boost your way into few-shot learning for image classification.

Random Erasing ⭐ 697

Random Erasing Data Augmentation. Experiments on CIFAR10, CIFAR100 and Fashion-MNIST

Repository of Vision Transformer with Deformable Attention (CVPR2022) and DAT++: Spatially Dynamic Vision Transformerwith Deformable Attention

Pytorch Best Practice ⭐ 499

A Guidance on PyTorch Coding Style Based on Kaggle Dogs vs. Cats

Pytorch Image Classification ⭐ 475

Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.

Neural Backed Decision Trees ⭐ 445

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

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

Residualmaskingnetwork ⭐ 387

ICPR 2020: Facial Expression Recognition using Residual Masking Network

Machine Learning Guide ⭐ 366

Machine learning Guide. Learn all about Machine Learning Tools, Libraries, Frameworks, Large Language Models (LLMs), and Training 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

Nfnets Pytorch ⭐ 316

NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/

Neural Pipeline ⭐ 314

Neural networks training pipeline based on PyTorch

Sparsify ⭐ 310

ML model optimization product to accelerate inference.

Modelassistant ⭐ 268

Seeed SenseCraft Model Assistant is an open-source project focused on embedded AI. 🔥🔥🔥

Trojanzoo ⭐ 260

TrojanZoo provides a universal pytorch platform to conduct security researches (especially backdoor attacks/defenses) of image classification in deep learning.

Nodeformer ⭐ 244

The official implementation of NeurIPS22 spotlight paper "NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classification"

Wildcat.pytorch ⭐ 227

PyTorch implementation of "WILDCAT: Weakly Supervised Learning of Deep ConvNets for Image Classification, Pointwise Localization and Segmentation", CVPR 2017

Pytorch_fine_tuning_tutorial ⭐ 226

A short tutorial on performing fine tuning or transfer learning in PyTorch.

(CVPR 2020) Block-wisely Supervised Neural Architecture Search with Knowledge Distillation

Parametric Contrastive Learning ⭐ 220

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

A multi-functional library for full-stack Deep Learning. Simplifies Model Building, API development, and Model Deployment.

Mobilenetv2 Pytorch ⭐ 214

Impementation of MobileNetV2 in pytorch

Difformer ⭐ 204

The official implementation for ICLR23 spotlight paper "DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusion"

Nsfw Resnet ⭐ 195

🔥🔥NSFW implement in pytorch（色情图&性感图识别，本程序经过了线上大数据集测试，性能优异效果良好）🔥🔥

Machinelearning Ai ⭐ 184

This repository contains all the work that I regularly did and studied from Medium blogs, several research papers, and other Repos (related/unrelated to the research papers).

Ncnn Models ⭐ 177

awesome AI models with NCNN, and how they were converted ✨✨✨

Hugsvision ⭐ 170

HugsVision is a easy to use huggingface wrapper for state-of-the-art computer vision

Iresnet ⭐ 157

Improved Residual Networks (https://arxiv.org/pdf/2004.04989.pdf)

AiTLAS implements state-of-the-art AI methods for exploratory and predictive analysis of satellite images.

Autoclint ⭐ 150

A specially designed light version of Fast AutoAugment

'NKD and USKD' (ICCV 2023) and 'ViTKD'

Food Recognition ⭐ 143

🍔🍟🍗 Food analysis baseline with Theseus. Integrate object detection, image classification and multi-class semantic segmentation 🍞🍖🍕

Remo Python ⭐ 131

🐰 Python lib for remo - the app for annotations and images management in Computer Vision

AgML is a centralized framework for agricultural machine learning. AgML provides access to public agricultural datasets for common agricultural deep learning tasks, with standard benchmarks and pretrained models, as well the ability to generate synthetic data and annotations.

Iciar2018 ⭐ 124

Two-Stage Convolutional Neural Network for Breast Cancer Histology Image Classification. ICIAR 2018 Grand Challenge on BreAst Cancer Histology images (BACH)

Masked Generative Distillation (ECCV 2022)

Pytorch_classifiers ⭐ 123

Almost any Image classification problem using pytorch

PyTorch reimplementation of the paper "MaxViT: Multi-Axis Vision Transformer" [arXiv 2022].

Pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"

Deep_learning_in_python_2018 ⭐ 114

Deep Learning workshop including image classification, face recognition, Object detection, language modelling, image captioning and neural machine translation.

Imageclassification Pytorch ⭐ 114

Ws_dan_pytorch ⭐ 112

PyTorch Implementation Of WS-DAN(See Better Before Looking Closer: Weakly Supervised Data Augmentation Network for Fine-Grained Visual Classification)

Convolution Vision Transformers ⭐ 110

PyTorch Implementation of CvT: Introducing Convolutions to Vision Transformers

Deep Learning Image Classification Models Based Cnn Or Attention ⭐ 104

This project organizes classic images classification neural networks based on convolution or attention, and writes training and inference python scripts

Moganet ⭐ 100

[ICLR 2024] MogaNet: Efficient Multi-order Gated Aggregation Network

Pytorch Learning Notes ⭐ 88

pytorch learning tutorials

Protest Detection Violence Estimation ⭐ 87

Implementation of the model used in the paper Protest Activity Detection and Perceived Violence Estimation from Social Media Images (ACM Multimedia 2017)

Prompt Align ⭐ 87

[ICCV23] Prompt-aligned Gradient for Prompt Tuning

Scene Baseline ⭐ 85

PyTorch baseline for AI challenger Scene classification

Ceit Pytorch ⭐ 84

Implementation of Convolutional enhanced image Transformer

Torchflare ⭐ 83

TorchFlare is a simple, beginner-friendly, and easy-to-use PyTorch Framework train your models effortlessly.

Simplenet ⭐ 83

This repository contains the architectures, Models, logs, etc pertaining to the SimpleNet Paper (Lets keep it simple: Using simple architectures to outperform deeper architectures )

Pytorch Auto Augment ⭐ 82

PyTorch implementation of AutoAugment.

Prg4ssl Mnar ⭐ 72

"Towards Semi-supervised Learning with Non-random Missing Labels" by Yue Duan (ICCV 2023)

Pytorch Learners Tutorial ⭐ 71

PyTorch tutorial for learners

Awesome Computer Vision Resources ⭐ 70

a collection of computer vision projects&tools. 计算机视觉方向项目和工具集合。

Satellighte ⭐ 68

📡 PyTorch Lightning Implementations of Recent Satellite Image Classification !

Reference implementation for Blueprint Separable Convolutions (CVPR 2020)

Pytorch Cifar Model Hub ⭐ 65

Implementation of Conv-based and Vit-based networks designed for CIFAR.

Mobilenetv3 Pytorch ⭐ 64

Implementation of MobileNetV3 in pytorch

Sota Backbones ⭐ 64

A collection of SOTA Image Classification Models in PyTorch

Pytorch_notebooks ⭐ 64

A collection of PyTorch notebooks for learning and practicing deep learning

I will implement Fastai in each projects present in this repository.

Tfvaegan ⭐ 62

[ECCV 2020] Official Pytorch implementation for "Latent Embedding Feedback and Discriminative Features for Zero-Shot Classification". SOTA results for ZSL and GZSL

本仓库将使用Pytorch框架实现经典的图像分类网络、目标检测网络、图像分割网络，图像生成网络等，并

Image_classification_sota ⭐ 58

Training ImageNet / CIFAR models with sota strategies and fancy techniques such as ViT, KD, Rep, etc.

Erasing Attention Consistency ⭐ 57

Official implementation of the ECCV2022 paper: Learn From All: Erasing Attention Consistency for Noisy Label Facial Expression Recognition

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)

Network Pytorch (1,299)

Pytorch Computer Vision (1,230)

1-100 of 242 search results

Privacy | About | Terms | Follow Us On Twitter

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