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
33k stars
313 watching
5k forks
Language: Python
last commit: 3 months ago
Linked from 4 awesome lists
augmixconvnextdistributed-trainingdual-path-networksefficientnetimage-classificationimagenetmaxvitmixnetmobile-deep-learningmobilenet-v2mobilenetv3nfnetsnormalization-free-trainingpretrained-modelspretrained-weightspytorchrandaugmentresnetvision-transformer-models
josephmisiti/awesome-machine-learning
bharathgs/awesome-pytorch-list
amrzv/awesome-colab-notebooks
balavenkatesh3322/cv-pretrained-modelRelated projects:
| Repository | Description | Stars |
|---|---|---|
 wiedehopf/tar1090 | An improved web interface for ADS-B decoder data analysis | 1,314 |
 langnico/global-canopy-height-model | Develops a high-resolution canopy height model to estimate canopy top height anywhere on Earth | 145 |
 mprakhar/dsm2dtm | Converts satellite images into digital surface models with building height extraction | 15 |
 rwth-ebc/teaser | Tools for generating and simulating energy-efficient buildings in urban environments | 119 |
 cliansang/positioning-algorithms-for-uwb-matlab | Matlab implementation of various positioning algorithms for UWB localization systems | 92 |
 iwugermany/dibs---dynamic-iso-building-simulator | A simulation program for calculating energy consumption and greenhouse gas emissions in German non-residential buildings | 24 |
 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 | 42 |
 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 | 80 |
 mbrossar/ai-imu-dr | Develops an accurate method for dead reckoning of wheeled vehicles based only on IMU data | 823 |