awesome-image-classification
A curated list of deep learning image classification papers and codes
https://github.com/weiaicunzai/awesome-image-classification
Last synced: 10 days ago
JSON representation
Background
DLA
AmoebaNet
- https://arxiv.org/abs/1802.01548
- tensorflow-tpu : https://github.com/tensorflow/tpu/tree/master/models/official/amoeba_net
BAM
- https://arxiv.org/abs/1807.06514
- official-pytorch : https://github.com/Jongchan/attention-module
- unofficial-tensorflow : https://github.com/huyz1117/BAM
BiT
- https://arxiv.org/abs/1912.11370
- official-tensorflow: https://github.com/google-research/big_transfer
CBAM
- https://arxiv.org/abs/1807.06521
- unofficial-pytorch : https://github.com/luuuyi/CBAM.PyTorch
- unofficial-pytorch : https://github.com/elbuco1/CBAM
- unofficial-keras : https://github.com/kobiso/CBAM-keras
CondenseNet
- https://arxiv.org/abs/1711.09224
- official : https://github.com/ShichenLiu/CondenseNet
- unofficial-tensorflow : https://github.com/markdtw/condensenet-tensorflow
CRU-Net
CViT
CvT
DARTS
- https://arxiv.org/abs/1806.09055
- official : https://github.com/quark0/darts
- unofficial-pytorch : https://github.com/khanrc/pt.darts
- unofficial-tensorflow : https://github.com/NeroLoh/darts-tensorflow
DeepVit
DeiT
EfficientNet
- https://arxiv.org/abs/1905.11946
- official-tensorflow : https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet
- unofficial-pytorch : https://github.com/lukemelas/EfficientNet-PyTorch
FixResNeXt
Focal-T
GhostNet
hier
IGCV2
IGCV3
- https://arxiv.org/abs/1806.00178
- official : https://github.com/homles11/IGCV3
- unofficial-pytorch : https://github.com/xxradon/IGCV3-pytorch
- unofficial-tensorflow : https://github.com/ZHANG-SHI-CHANG/IGCV3
LIP-ResNet
MNasNet
- https://arxiv.org/abs/1807.11626
- unofficial-pytorch : https://github.com/AnjieZheng/MnasNet-PyTorch
- unofficial-caffe : https://github.com/LiJianfei06/MnasNet-caffe
- unofficial-MxNet : https://github.com/chinakook/Mnasnet.MXNet
- unofficial-keras : https://github.com/Shathe/MNasNet-Keras-Tensorflow
MobileNetV2
- https://arxiv.org/abs/1801.04381
- unofficial-keras : https://github.com/xiaochus/MobileNetV2
- unofficial-pytorch : https://github.com/Randl/MobileNetV2-pytorch
- unofficial-tensorflow : https://github.com/neuleaf/MobileNetV2
MobileNetV3
- https://arxiv.org/abs/1905.02244
- unofficial-pytorch : https://github.com/xiaolai-sqlai/mobilenetv3
- unofficial-pytorch : https://github.com/kuan-wang/pytorch-mobilenet-v3
- unofficial-pytorch : https://github.com/leaderj1001/MobileNetV3-Pytorch
- unofficial-pytorch : https://github.com/d-li14/mobilenetv3.pytorch
- unofficial-caffe : https://github.com/jixing0415/caffe-mobilenet-v3
- unofficial-keras : https://github.com/xiaochus/MobileNetV3
NasNet
- https://arxiv.org/abs/1707.07012
- keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/nasnet.py
- unofficial-keras : https://github.com/titu1994/Keras-NASNet
- unofficial-pytorch : https://github.com/wandering007/nasnet-pytorch
- unofficial-tensorflow : https://github.com/yeephycho/nasnet-tensorflow
NoisyStudent
- https://arxiv.org/abs/1911.04252
- official-tensorflow: https://github.com/google-research/noisystudent
- unofficial-pytorch: https://github.com/sally20921/NoisyStudent
PNasNet
- https://arxiv.org/abs/1712.00559
- tensorflow-slim : https://github.com/tensorflow/models/blob/master/research/slim/nets/nasnet/pnasnet.py
- unofficial-pytorch : https://github.com/chenxi116/PNASNet.pytorch
- unofficial-tensorflow : https://github.com/chenxi116/PNASNet.TF
ProxylessNAS
PSConv + ResNext101
PVT
PVTv2
RegNet
- https://arxiv.org/abs/2003.13678
- official-pytorch: https://github.com/facebookresearch/pycls
- unofficial-pytorch: https://github.com/d-li14/regnet.pytorch
Res2Net
- https://arxiv.org/abs/1904.01169
- official-pytorch : https://github.com/Res2Net
- unofficial-pytorch : https://github.com/4uiiurz1/pytorch-res2net
- unofficial-keras : https://github.com/fupiao1998/res2net-keras
SENet
- https://arxiv.org/abs/1709.01507
- official : https://github.com/hujie-frank/SENet
- unofficial-pytorch : https://github.com/moskomule/senet.pytorch
- unofficial-tensorflow : https://github.com/taki0112/SENet-Tensorflow
- unofficial-caffe : https://github.com/shicai/SENet-Caffe
- unofficial-mxnet : https://github.com/bruinxiong/SENet.mxnet
ShuffleNet
- https://arxiv.org/abs/1707.01083
- unofficial-tensorflow : https://github.com/MG2033/ShuffleNet
- unofficial-pytorch : https://github.com/jaxony/ShuffleNet
- unofficial-caffe : https://github.com/farmingyard/ShuffleNet
- unofficial-keras : https://github.com/scheckmedia/keras-shufflenet
ShuffleNetV2
- https://arxiv.org/abs/1807.11164
- unofficial-pytorch : https://github.com/Randl/ShuffleNetV2-pytorch
- unofficial-keras : https://github.com/opconty/keras-shufflenetV2
- unofficial-caff2: https://github.com/wolegechu/ShuffleNetV2.Caffe2
SKNet
T2T
TNT
Twins
ViL
ViT
Papers&Codes
Block-QNN
CRU-Net
- https://arxiv.org/abs/1703.02180
- official : https://github.com/cypw/CRU-Net
- unofficial-mxnet : https://github.com/bruinxiong/Modified-CRUNet-and-Residual-Attention-Network.mxnet
DenseNet
- https://arxiv.org/abs/1608.06993
- official : https://github.com/liuzhuang13/DenseNet
- unofficial-keras : https://github.com/titu1994/DenseNet
- unofficial-caffe : https://github.com/shicai/DenseNet-Caffe
- unofficial-pytorch : https://github.com/YixuanLi/densenet-tensorflow
- unofficial-pytorch : https://github.com/bamos/densenet.pytorch
- unofficial-keras : https://github.com/flyyufelix/DenseNet-Keras
DPN
- https://arxiv.org/abs/1707.01629
- official : https://github.com/cypw/DPNs
- unoffical-keras : https://github.com/titu1994/Keras-DualPathNetworks
- unofficial-pytorch : https://github.com/oyam/pytorch-DPNs
- unofficial-pytorch : https://github.com/rwightman/pytorch-dpn-pretrained
FractalNet
- https://arxiv.org/abs/1605.07648
- unofficial-caffe : https://github.com/gustavla/fractalnet
- unofficial-keras : https://github.com/snf/keras-fractalnet
- unofficial-tensorflow : https://github.com/tensorpro/FractalNet
GeNet
GoogleNet
- https://arxiv.org/abs/1409.4842
- unofficial-tensorflow : https://github.com/conan7882/GoogLeNet-Inception
- unofficial-caffe : https://github.com/lim0606/caffe-googlenet-bn
IGCV1
- https://arxiv.org/abs/1707.02725
- official : https://github.com/hellozting/InterleavedGroupConvolutions
Inceptionv3
- https://arxiv.org/abs/1512.00567
- torchvision : https://github.com/pytorch/vision/blob/master/torchvision/models/inception.py
- keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/inception_v3.py
Inceptionv4 && Inception-ResNetv2
- https://arxiv.org/abs/1602.07261
- unofficial-keras : https://github.com/kentsommer/keras-inceptionV4
- unofficial-keras : https://github.com/titu1994/Inception-v4
- unofficial-keras : https://github.com/yuyang-huang/keras-inception-resnet-v2
MetaQNN
MobileNet
- https://arxiv.org/abs/1704.04861
- keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/mobilenet.py
- unofficial-tensorflow : https://github.com/Zehaos/MobileNet
- unofficial-caffe : https://github.com/shicai/MobileNet-Caffe
- unofficial-pytorch : https://github.com/marvis/pytorch-mobilenet
PolyNet
PreActResNet
- https://arxiv.org/abs/1603.05027
- facebook-torch : https://github.com/facebook/fb.resnet.torch/blob/master/models/preresnet.lua
- unoffical-pytorch : https://github.com/kuangliu/pytorch-cifar/blob/master/models/preact_resnet.py
- official : https://github.com/KaimingHe/resnet-1k-layers
- unoffical-mxnet : https://github.com/tornadomeet/ResNet
PReLU-nets
PyramidNet
- https://arxiv.org/abs/1610.02915
- official : https://github.com/jhkim89/PyramidNet
- unofficial-pytorch : https://github.com/dyhan0920/PyramidNet-PyTorch
Residual Attention Network
- https://arxiv.org/abs/1704.06904
- official : https://github.com/fwang91/residual-attention-network
- unofficial-pytorch : https://github.com/tengshaofeng/ResidualAttentionNetwork-pytorch
- unofficial-gluon : https://github.com/PistonY/ResidualAttentionNetwork
- unofficial-keras : https://github.com/koichiro11/residual-attention-network
ResNet
- https://arxiv.org/abs/1512.03385
- torchvision : https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py
- keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/resnet.py
- unofficial-keras : https://github.com/raghakot/keras-resnet
- unofficial-tensorflow : https://github.com/ry/tensorflow-resnet
- facebook-torch : https://github.com/facebook/fb.resnet.torch
ResNext
- https://arxiv.org/abs/1611.05431
- keras-applications : https://github.com/keras-team/keras-applications/blob/master/keras_applications/resnext.py
- official : https://github.com/facebookresearch/ResNeXt
- unofficial-pytorch : https://github.com/prlz77/ResNeXt.pytorch
- unofficial-keras : https://github.com/titu1994/Keras-ResNeXt
- unofficial-tensorflow : https://github.com/taki0112/ResNeXt-Tensorflow
- unofficial-tensorflow : https://github.com/wenxinxu/ResNeXt-in-tensorflow
RiR
- https://arxiv.org/abs/1603.08029
- unofficial-tensorflow : https://github.com/SunnerLi/RiR-Tensorflow
- unofficial-chainer : https://github.com/nutszebra/resnet_in_resnet
SqueezeNet
- https://arxiv.org/abs/1602.07360
- torchvision : https://github.com/pytorch/vision/blob/master/torchvision/models/squeezenet.py
- unofficial-caffe : https://github.com/DeepScale/SqueezeNet
- unofficial-keras : https://github.com/rcmalli/keras-squeezenet
- unofficial-caffe : https://github.com/songhan/SqueezeNet-Residual
- unofficial-caffe : https://github.com/DeepScale/SqueezeNet