awesome-image-classification

Image Classification Library

A curated list of deep learning image classification papers and codes to help beginners learn computer vision.

A curated list of deep learning image classification papers and codes

GitHub

3k stars
85 watching
600 forks
last commit: over 2 years ago
Linked from 1 awesome list

awesomeawesome-listcomputer-visiondeep-learningimage-classificationpapers

Awesome - Image Classification / Performance Table

rejected by ICLR2017 arXiv2017( )

Awesome - Image Classification / Papers&Codes / VGG

https://arxiv.org/abs/1409.1556 pdf:
torchvision : https://github.com/pytorch/vision/blob/master/torchvision/models/vgg.py 16,251 6 days ago code:
keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/vgg16.py 2,000 almost 3 years ago code:
keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/vgg19.py 2,000 almost 3 years ago code:

Awesome - Image Classification / Papers&Codes / GoogleNet

https://arxiv.org/abs/1409.4842 pdf:
unofficial-tensorflow : https://github.com/conan7882/GoogLeNet-Inception 282 over 4 years ago code:
unofficial-caffe : https://github.com/lim0606/caffe-googlenet-bn 131 almost 8 years ago code:

Awesome - Image Classification / Papers&Codes / PReLU-nets

https://arxiv.org/abs/1502.01852 pdf:
unofficial-chainer : https://github.com/nutszebra/prelu_net 12 almost 8 years ago code:

Awesome - Image Classification / Papers&Codes / ResNet

https://arxiv.org/abs/1512.03385 pdf:
facebook-torch : https://github.com/facebook/fb.resnet.torch 2,296 about 2 years ago code:
torchvision : https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py 16,251 6 days ago code:
keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/resnet.py 2,000 almost 3 years ago code:
unofficial-keras : https://github.com/raghakot/keras-resnet 1,386 almost 4 years ago code:
unofficial-tensorflow : https://github.com/ry/tensorflow-resnet 1,658 over 6 years ago code:

Awesome - Image Classification / Papers&Codes / PreActResNet

https://arxiv.org/abs/1603.05027 pdf:
facebook-torch : https://github.com/facebook/fb.resnet.torch/blob/master/models/preresnet.lua 2,296 about 2 years ago code:
official : https://github.com/KaimingHe/resnet-1k-layers 906 over 7 years ago code:
unoffical-pytorch : https://github.com/kuangliu/pytorch-cifar/blob/master/models/preact_resnet.py 5,991 over 1 year ago code:
unoffical-mxnet : https://github.com/tornadomeet/ResNet 556 over 6 years ago code:

Awesome - Image Classification / Papers&Codes / Inceptionv3

https://arxiv.org/abs/1512.00567 pdf:
torchvision : https://github.com/pytorch/vision/blob/master/torchvision/models/inception.py 16,251 6 days ago code:
keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/inception_v3.py 2,000 almost 3 years ago code:

Awesome - Image Classification / Papers&Codes / Inceptionv4 && Inception-ResNetv2

https://arxiv.org/abs/1602.07261 pdf:
unofficial-keras : https://github.com/kentsommer/keras-inceptionV4 471 about 7 years ago code:
unofficial-keras : https://github.com/titu1994/Inception-v4 386 about 7 years ago code:
unofficial-keras : https://github.com/yuyang-huang/keras-inception-resnet-v2 180 over 3 years ago code:

Awesome - Image Classification / Papers&Codes / RiR

https://arxiv.org/abs/1603.08029 pdf:
unofficial-tensorflow : https://github.com/SunnerLi/RiR-Tensorflow 4 about 7 years ago code:
unofficial-chainer : https://github.com/nutszebra/resnet_in_resnet 3 almost 8 years ago code:

Awesome - Image Classification / Papers&Codes / Stochastic Depth ResNet

https://arxiv.org/abs/1603.09382 pdf:
unofficial-torch : https://github.com/yueatsprograms/Stochastic_Depth 477 over 6 years ago code:
unofficial-chainer : https://github.com/yasunorikudo/chainer-ResDrop 40 over 8 years ago code:
unofficial-keras : https://github.com/dblN/stochastic_depth_keras 139 over 4 years ago code:

Awesome - Image Classification / Papers&Codes / WRN

https://arxiv.org/abs/1605.07146 pdf:
official : https://github.com/szagoruyko/wide-residual-networks 1,298 over 5 years ago code:
unofficial-pytorch : https://github.com/xternalz/WideResNet-pytorch 333 over 3 years ago code:
unofficial-keras : https://github.com/asmith26/wide_resnets_keras 138 10 months ago code:
unofficial-pytorch : https://github.com/meliketoy/wide-resnet.pytorch 461 almost 5 years ago code:

Awesome - Image Classification / Papers&Codes / SqueezeNet

https://arxiv.org/abs/1602.07360 pdf:
torchvision : https://github.com/pytorch/vision/blob/master/torchvision/models/squeezenet.py 16,251 6 days ago code:
unofficial-caffe : https://github.com/DeepScale/SqueezeNet 2,176 over 6 years ago code:
unofficial-keras : https://github.com/rcmalli/keras-squeezenet 404 over 1 year ago code:
unofficial-caffe : https://github.com/songhan/SqueezeNet-Residual 154 over 5 years ago code:

Awesome - Image Classification / Papers&Codes / GeNet

https://arxiv.org/abs/1703.01513 pdf:
unofficial-tensorflow : https://github.com/aqibsaeed/Genetic-CNN 218 over 2 years ago code:

Awesome - Image Classification / Papers&Codes / MetaQNN

https://arxiv.org/abs/1611.02167 pdf:
official : https://github.com/bowenbaker/metaqnn 137 over 7 years ago code:

Awesome - Image Classification / Papers&Codes / PyramidNet

https://arxiv.org/abs/1610.02915 pdf:
official : https://github.com/jhkim89/PyramidNet 129 about 7 years ago code:
unofficial-pytorch : https://github.com/dyhan0920/PyramidNet-PyTorch 273 over 4 years ago code:

Awesome - Image Classification / Papers&Codes / DenseNet

https://arxiv.org/abs/1608.06993 pdf:
official : https://github.com/liuzhuang13/DenseNet 4,740 11 months ago code:
unofficial-keras : https://github.com/titu1994/DenseNet 708 over 4 years ago code:
unofficial-caffe : https://github.com/shicai/DenseNet-Caffe 357 about 5 years ago code:
unofficial-tensorflow : https://github.com/YixuanLi/densenet-tensorflow 573 over 5 years ago code:
unofficial-pytorch : https://github.com/YixuanLi/densenet-tensorflow 573 over 5 years ago code:
unofficial-pytorch : https://github.com/bamos/densenet.pytorch 832 over 6 years ago code:
unofficial-keras : https://github.com/flyyufelix/DenseNet-Keras 568 about 5 years ago code:

Awesome - Image Classification / Papers&Codes / FractalNet

https://arxiv.org/abs/1605.07648 pdf:
unofficial-caffe : https://github.com/gustavla/fractalnet 151 about 7 years ago code:
unofficial-keras : https://github.com/snf/keras-fractalnet 155 about 7 years ago code:
unofficial-tensorflow : https://github.com/tensorpro/FractalNet 48 almost 8 years ago code:

Awesome - Image Classification / Papers&Codes / ResNext

https://arxiv.org/abs/1611.05431 pdf:
official : https://github.com/facebookresearch/ResNeXt 1,906 almost 5 years ago code:
keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/resnext.py 2,000 almost 3 years ago code:
unofficial-pytorch : https://github.com/prlz77/ResNeXt.pytorch 508 over 4 years ago code:
unofficial-keras : https://github.com/titu1994/Keras-ResNeXt 224 over 3 years ago code:
unofficial-tensorflow : https://github.com/taki0112/ResNeXt-Tensorflow 158 over 6 years ago code:
unofficial-tensorflow : https://github.com/wenxinxu/ResNeXt-in-tensorflow 78 almost 7 years ago code:

Awesome - Image Classification / Papers&Codes / IGCV1

https://arxiv.org/abs/1707.02725 pdf:
official : https://github.com/hellozting/InterleavedGroupConvolutions 111 almost 7 years ago code

Awesome - Image Classification / Papers&Codes / Residual Attention Network

https://arxiv.org/abs/1704.06904 pdf:
official : https://github.com/fwang91/residual-attention-network 551 over 7 years ago code:
unofficial-pytorch : https://github.com/tengshaofeng/ResidualAttentionNetwork-pytorch 680 about 6 years ago code:
unofficial-gluon : https://github.com/PistonY/ResidualAttentionNetwork 107 over 5 years ago code:
unofficial-keras : https://github.com/koichiro11/residual-attention-network 94 over 6 years ago code:

Awesome - Image Classification / Papers&Codes / Xception

https://arxiv.org/abs/1610.02357 pdf:
unofficial-pytorch : https://github.com/jfzhang95/pytorch-deeplab-xception/blob/master/modeling/backbone/xception.py 2,910 4 months ago code:
unofficial-tensorflow : https://github.com/kwotsin/TensorFlow-Xception 207 about 5 years ago code:
unofficial-caffe : https://github.com/yihui-he/Xception-caffe 45 7 months ago code:
unofficial-pytorch : https://github.com/tstandley/Xception-PyTorch 250 almost 2 years ago code:
keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/xception.py 2,000 almost 3 years ago code:

Awesome - Image Classification / Papers&Codes / MobileNet

https://arxiv.org/abs/1704.04861 pdf:
unofficial-tensorflow : https://github.com/Zehaos/MobileNet 1,617 about 7 years ago code:
unofficial-caffe : https://github.com/shicai/MobileNet-Caffe 1,259 over 3 years ago code:
unofficial-pytorch : https://github.com/marvis/pytorch-mobilenet code:
keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/mobilenet.py 2,000 almost 3 years ago code:

Awesome - Image Classification / Papers&Codes / PolyNet

https://arxiv.org/abs/1611.05725 pdf:
official : https://github.com/open-mmlab/polynet 82 over 7 years ago code:

Awesome - Image Classification / Papers&Codes / DPN

https://arxiv.org/abs/1707.01629 pdf:
official : https://github.com/cypw/DPNs 537 over 5 years ago code:
unoffical-keras : https://github.com/titu1994/Keras-DualPathNetworks 114 almost 5 years ago code:
unofficial-pytorch : https://github.com/oyam/pytorch-DPNs 90 about 7 years ago code:
unofficial-pytorch : https://github.com/rwightman/pytorch-dpn-pretrained 206 over 5 years ago code:

Awesome - Image Classification / Papers&Codes / Block-QNN

https://arxiv.org/abs/1708.05552 pdf:

Awesome - Image Classification / Papers&Codes / CRU-Net

https://arxiv.org/abs/1703.02180 pdf:
official : https://github.com/cypw/CRU-Net 45 over 7 years ago code
unofficial-mxnet : https://github.com/bruinxiong/Modified-CRUNet-and-Residual-Attention-Network.mxnet 67 about 6 years ago code

Awesome - Image Classification / DLA

https://arxiv.org/abs/1707.06484 pdf:
official-pytorch: https://github.com/ucbdrive/dla 434 almost 3 years ago code:

Awesome - Image Classification / DLA / ShuffleNet

https://arxiv.org/abs/1707.01083 pdf:
unofficial-tensorflow : https://github.com/MG2033/ShuffleNet 384 almost 6 years ago code:
unofficial-pytorch : https://github.com/jaxony/ShuffleNet 294 almost 7 years ago code:
unofficial-caffe : https://github.com/farmingyard/ShuffleNet 448 about 6 years ago code:
unofficial-keras : https://github.com/scheckmedia/keras-shufflenet 77 about 4 years ago code:

Awesome - Image Classification / DLA / CondenseNet

https://arxiv.org/abs/1711.09224 pdf:
official : https://github.com/ShichenLiu/CondenseNet 694 about 5 years ago code:
unofficial-tensorflow : https://github.com/markdtw/condensenet-tensorflow 29 almost 7 years ago code:

Awesome - Image Classification / DLA / NasNet

https://arxiv.org/abs/1707.07012 pdf:
unofficial-keras : https://github.com/titu1994/Keras-NASNet 199 over 6 years ago code:
keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/nasnet.py 2,000 almost 3 years ago code:
unofficial-pytorch : https://github.com/wandering007/nasnet-pytorch 75 almost 4 years ago code:
unofficial-tensorflow : https://github.com/yeephycho/nasnet-tensorflow 136 over 3 years ago code:

Awesome - Image Classification / DLA / MobileNetV2

https://arxiv.org/abs/1801.04381 pdf:
unofficial-keras : https://github.com/xiaochus/MobileNetV2 320 over 5 years ago code:
unofficial-pytorch : https://github.com/Randl/MobileNetV2-pytorch 271 about 6 years ago code:
unofficial-tensorflow : https://github.com/neuleaf/MobileNetV2 150 over 5 years ago code:

Awesome - Image Classification / DLA / IGCV2

https://arxiv.org/abs/1804.06202 pdf:

Awesome - Image Classification / DLA / hier

https://arxiv.org/abs/1711.00436 pdf:

Awesome - Image Classification / DLA / PNasNet

https://arxiv.org/abs/1712.00559 pdf:
tensorflow-slim : https://github.com/tensorflow/models/blob/master/research/slim/nets/nasnet/pnasnet.py 77,177 6 days ago code:
unofficial-pytorch : https://github.com/chenxi116/PNASNet.pytorch 317 over 2 years ago code:
unofficial-tensorflow : https://github.com/chenxi116/PNASNet.TF 102 almost 6 years ago code:

Awesome - Image Classification / DLA / AmoebaNet

https://arxiv.org/abs/1802.01548 pdf:
tensorflow-tpu : https://github.com/tensorflow/tpu/tree/master/models/official/amoeba_net 5,213 3 months ago code:

Awesome - Image Classification / DLA / SENet

https://arxiv.org/abs/1709.01507 pdf:
official : https://github.com/hujie-frank/SENet 3,394 over 5 years ago code:
unofficial-pytorch : https://github.com/moskomule/senet.pytorch 2,285 over 3 years ago code:
unofficial-tensorflow : https://github.com/taki0112/SENet-Tensorflow 756 over 6 years ago code:
unofficial-caffe : https://github.com/shicai/SENet-Caffe 169 almost 6 years ago code:
unofficial-mxnet : https://github.com/bruinxiong/SENet.mxnet 154 over 6 years ago code:

Awesome - Image Classification / DLA / ShuffleNetV2

https://arxiv.org/abs/1807.11164 pdf:
unofficial-pytorch : https://github.com/Randl/ShuffleNetV2-pytorch 191 over 2 years ago code:
unofficial-keras : https://github.com/opconty/keras-shufflenetV2 96 over 6 years ago code:
unofficial-pytorch : https://github.com/Bugdragon/ShuffleNet_v2_PyTorch code:
unofficial-caff2: https://github.com/wolegechu/ShuffleNetV2.Caffe2 25 over 6 years ago code:

Awesome - Image Classification / DLA / CBAM

https://arxiv.org/abs/1807.06521 pdf:
official-pytorch : https://github.com/Jongchan/attention-module 2,061 over 1 year ago code:
unofficial-pytorch : https://github.com/luuuyi/CBAM.PyTorch 1,337 over 1 year ago code:
unofficial-pytorch : https://github.com/elbuco1/CBAM 104 about 5 years ago code:
unofficial-keras : https://github.com/kobiso/CBAM-keras 361 almost 4 years ago code:

Awesome - Image Classification / DLA / IGCV3

https://arxiv.org/abs/1806.00178 pdf:
official : https://github.com/homles11/IGCV3 189 about 6 years ago code:
unofficial-pytorch : https://github.com/xxradon/IGCV3-pytorch 19 over 6 years ago code:
unofficial-tensorflow : https://github.com/ZHANG-SHI-CHANG/IGCV3 2 over 6 years ago code:

Awesome - Image Classification / DLA / BAM

https://arxiv.org/abs/1807.06514 pdf:
official-pytorch : https://github.com/Jongchan/attention-module 2,061 over 1 year ago code:
unofficial-tensorflow : https://github.com/huyz1117/BAM 12 almost 6 years ago code:

Awesome - Image Classification / DLA / MNasNet

https://arxiv.org/abs/1807.11626 pdf:
unofficial-pytorch : https://github.com/AnjieZheng/MnasNet-PyTorch 271 over 6 years ago code:
unofficial-caffe : https://github.com/LiJianfei06/MnasNet-caffe 52 about 6 years ago code:
unofficial-MxNet : https://github.com/chinakook/Mnasnet.MXNet 59 almost 6 years ago code:
unofficial-keras : https://github.com/Shathe/MNasNet-Keras-Tensorflow 99 over 4 years ago code:

Awesome - Image Classification / DLA / SKNet

https://arxiv.org/abs/1903.06586 pdf:
official : https://github.com/implus/SKNet 593 over 5 years ago code:

Awesome - Image Classification / DLA / DARTS

https://arxiv.org/abs/1806.09055 pdf:
official : https://github.com/quark0/darts 3,919 almost 4 years ago code:
unofficial-pytorch : https://github.com/khanrc/pt.darts 441 over 2 years ago code:
unofficial-tensorflow : https://github.com/NeroLoh/darts-tensorflow 73 almost 6 years ago code:

Awesome - Image Classification / DLA / ProxylessNAS

https://arxiv.org/abs/1812.00332 pdf:
official : https://github.com/mit-han-lab/ProxylessNAS 1,425 3 months ago code:

Awesome - Image Classification / DLA / MobileNetV3

https://arxiv.org/abs/1905.02244 pdf:
unofficial-pytorch : https://github.com/xiaolai-sqlai/mobilenetv3 1,650 over 1 year ago code:
unofficial-pytorch : https://github.com/kuan-wang/pytorch-mobilenet-v3 779 over 5 years ago code:
unofficial-pytorch : https://github.com/leaderj1001/MobileNetV3-Pytorch 292 over 5 years ago code:
unofficial-pytorch : https://github.com/d-li14/mobilenetv3.pytorch 521 over 1 year ago code:
unofficial-caffe : https://github.com/jixing0415/caffe-mobilenet-v3 130 over 5 years ago code:
unofficial-keras : https://github.com/xiaochus/MobileNetV3 238 over 4 years ago code:

Awesome - Image Classification / DLA / Res2Net

https://arxiv.org/abs/1904.01169 pdf:
unofficial-pytorch : https://github.com/4uiiurz1/pytorch-res2net 112 over 5 years ago code:
unofficial-keras : https://github.com/fupiao1998/res2net-keras 16 over 5 years ago code:
official-pytorch : https://github.com/Res2Net code:

Awesome - Image Classification / DLA / LIP-ResNet

https://arxiv.org/abs/1908.04156 pdf:
official-pytorch : https://github.com/sebgao/LIP 220 about 3 years ago code:

Awesome - Image Classification / DLA / EfficientNet

https://arxiv.org/abs/1905.11946 pdf:
unofficial-pytorch : https://github.com/lukemelas/EfficientNet-PyTorch 7,921 over 2 years ago code:
official-tensorflow : https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet 5,213 3 months ago code:

Awesome - Image Classification / DLA / FixResNeXt

https://arxiv.org/abs/1906.06423 pdf:
official-pytorch : https://github.com/facebookresearch/FixRes 1,033 over 3 years ago code:

Awesome - Image Classification / DLA / BiT

https://arxiv.org/abs/1912.11370 pdf:
official-tensorflow: https://github.com/google-research/big_transfer 1,513 4 months ago code:

Awesome - Image Classification / DLA / PSConv + ResNext101

https://arxiv.org/abs/2007.06191 pdf:
https://github.com/d-li14/PSConv 174 over 4 years ago code:

Awesome - Image Classification / DLA / NoisyStudent

https://arxiv.org/abs/1911.04252 pdf:
official-tensorflow: https://github.com/google-research/noisystudent 753 over 3 years ago code:
unofficial-pytorch: https://github.com/sally20921/NoisyStudent code:

Awesome - Image Classification / DLA / RegNet

https://arxiv.org/abs/2003.13678 pdf:
official-pytorch: https://github.com/facebookresearch/pycls 2,138 8 months ago code:
unofficial-pytorch: https://github.com/d-li14/regnet.pytorch 68 over 3 years ago code:

Awesome - Image Classification / DLA / GhostNet

https://arxiv.org/abs/1911.11907 pdf:
official-pytorch: https://github.com/huawei-noah/ghostnet 4,054 4 months ago code:

Awesome - Image Classification / DLA / ViT

https://arxiv.org/abs/2010.11929 pdf:
official-tensorflow: https://github.com/google-research/vision_transformer 10,450 6 months ago code:
unofficial-pytorch: https://github.com/jeonsworld/ViT-pytorch 1,940 over 2 years ago code:

Awesome - Image Classification / DLA / DeiT

https://arxiv.org/abs/2012.12877 pdf:
official-pytorch: https://github.com/facebookresearch/deit 4,064 8 months ago code:

Awesome - Image Classification / DLA / PVT

https://arxiv.org/abs/2102.12122 pdf:
official-pytorch: https://github.com/whai362/PVT 1,728 about 2 years ago code:

Awesome - Image Classification / DLA / T2T

https://arxiv.org/abs/2101.11986 pdf:
official-pytorch: https://github.com/yitu-opensource/T2T-ViT 1,148 about 1 year ago code:

Awesome - Image Classification / DLA / DeepVit

https://arxiv.org/abs/2103.11886 pdf:
official-pytorch: https://github.com/zhoudaquan/dvit_repo 136 almost 3 years ago code:

Awesome - Image Classification / DLA / ViL

https://arxiv.org/abs/2103.15358 pdf:
official-pytorch: https://github.com/microsoft/vision-longformer 241 over 2 years ago code:

Awesome - Image Classification / DLA / TNT

https://arxiv.org/abs/2103.00112 pdf:
https://github.com/huawei-noah/CV-Backbones 4,054 4 months ago code:

Awesome - Image Classification / DLA / CvT

https://arxiv.org/abs/2103.15808 pdf:
https://github.com/microsoft/CvT 555 over 1 year ago code:

Awesome - Image Classification / DLA / CViT

https://arxiv.org/abs/2103.14899 pdf:
https://github.com/IBM/CrossViT 352 almost 3 years ago code:

Awesome - Image Classification / DLA / Focal-T

https://arxiv.org/abs/2107.00641 pdf:
https://github.com/microsoft/Focal-Transformer 545 over 2 years ago code:

Awesome - Image Classification / DLA / Twins

https://arxiv.org/abs/2104.13840 pdf:
https://git.io/Twins code:

Awesome - Image Classification / DLA / PVTv2

https://arxiv.org/abs/2106.13797 pdf:
official-pytorch: https://github.com/whai362/PVT 1,728 about 2 years ago code:

Backlinks from these awesome lists:

More related projects: