pytorch-image-models

Image classifiers

A comprehensive collection of pre-trained PyTorch models for image classification tasks

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

GitHub

32k stars
313 watching
5k forks
Language: Python
last commit: 7 days ago
Linked from 4 awesome lists

augmixconvnextdistributed-trainingdual-path-networksefficientnetimage-classificationimagenetmaxvitmixnetmobile-deep-learningmobilenet-v2mobilenetv3nfnetsnormalization-free-trainingpretrained-modelspretrained-weightspytorchrandaugmentresnetvision-transformer-models

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
wiedehopf/tar1090 An improved web interface for ADS-B decoder data analysis 1,297
langnico/global-canopy-height-model Develops a high-resolution canopy height model to estimate canopy top height anywhere on Earth 138
mprakhar/dsm2dtm Converts satellite images into digital surface models with building height extraction 15
rwth-ebc/teaser A tool for simulating energy efficiency in buildings at urban scales 116
cliansang/positioning-algorithms-for-uwb-matlab Matlab implementation of various positioning algorithms for UWB localization systems 86
iwugermany/dibs---dynamic-iso-building-simulator A simulation program for calculating energy consumption and greenhouse gas emissions in German non-residential buildings 23
ofmooseandmen/jord-rs Geographical position calculations using spherical geometry and ellipsoidal models. 9
locationtech/geowave Provides geospatial and temporal indexing on top of various data stores and databases 502
db47h/decimal An arbitrary-precision decimal floating-point arithmetic package for Go 41
aws-samples/reinvent2015-dev309 Provides sample code examples for processing large-scale metrics in Ruby using Hadoop and Amazon Redshift 7
kevalpatel2106/collision-detector-android-things Calculates the distance of an object using ultrasonic ranging sensor HC-SR04 with Android Things 10
berthin/cerebro-base2x16 Converts numbers between decimal, hexadecimal, and binary formats 1
emf-creaf/meteoland Utilities to estimate daily weather variables at any position over complex terrains 10
pyrocko/kite Software for processing InSAR surface displacement data in support of earthquake modeling 79
mbrossar/ai-imu-dr Develops an accurate method for dead reckoning of wheeled vehicles based only on IMU data 810