pytorch-image-models

Image classification models

A collection of pre-trained PyTorch models and optimizers 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

33k stars
313 watching
5k forks
Language: Python
last commit: almost 2 years 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:

RepositoryDescriptionStars
wiedehopf/tar1090An improved web interface for ADS-B decoder data analysis1,314
langnico/global-canopy-height-modelDevelops a high-resolution canopy height model to estimate canopy top height anywhere on Earth145
mprakhar/dsm2dtmConverts satellite images into digital surface models with building height extraction15
rwth-ebc/teaserTools for generating and simulating energy-efficient buildings in urban environments119
cliansang/positioning-algorithms-for-uwb-matlabMatlab implementation of various positioning algorithms for UWB localization systems92
iwugermany/dibs---dynamic-iso-building-simulatorA simulation program for calculating energy consumption and greenhouse gas emissions in German non-residential buildings24
ofmooseandmen/jord-rsGeographical position calculations using spherical geometry and ellipsoidal models.9
locationtech/geowaveProvides geospatial and temporal indexing on top of various data stores and databases502
db47h/decimalAn arbitrary-precision decimal floating-point arithmetic package for Go42
aws-samples/reinvent2015-dev309Provides sample code examples for processing large-scale metrics in Ruby using Hadoop and Amazon Redshift7
kevalpatel2106/collision-detector-android-thingsCalculates the distance of an object using ultrasonic ranging sensor HC-SR04 with Android Things10
berthin/cerebro-base2x16Converts numbers between decimal, hexadecimal, and binary formats1
emf-creaf/meteolandUtilities to estimate daily weather variables at any position over complex terrains10
pyrocko/kiteSoftware for processing InSAR surface displacement data in support of earthquake modeling80
mbrossar/ai-imu-drDevelops an accurate method for dead reckoning of wheeled vehicles based only on IMU data823