紅色石頭的個人網站:

紅色石頭的個人博客-機器學習、深度學習之路?

redstonewill.com
圖標

今天給大家介紹自 2014 年以來,計算機視覺 CV 領域圖像分類方向文獻和代碼的超全總結和列表!總共涉及 36 種 ConvNet 模型。該 GitHub 項目作者是 weiaicunzai,項目地址是:

weiaicunzai/awesome-image-classification?

github.com
圖標

背景

我相信圖像識別是深入到其它機器視覺領域一個很好的起點,特別是對於剛剛入門深度學習的人來說。當我初學 CV 時,犯了很多錯。我當時非常希望有人能告訴我應該從哪一篇論文開始讀起。到目前為止,似乎還沒有一個像 deep-learning-object-detection 這樣的 GitHub 項目。因此,我決定建立一個 GitHub 項目,列出深入學習中關於圖像分類的論文和代碼,以幫助其他人。

對於學習路線,我的個人建議是,對於那些剛入門深度學習的人,可以試著從 vgg 開始,然後是 googlenet、resnet,之後可以自由地繼續閱讀列出的其它論文或切換到其它領域。

性能表

基於簡化的目的,我只從論文中列舉出在 ImageNet 上準確率最高的 top1 和 top5。注意,這並不一定意味著準確率越高,一個網路就比另一個網路更好。因為有些網路專註於降低模型複雜性而不是提高準確性,或者有些論文只給出 ImageNet 上的 single crop results,而另一些則給出模型融合或 multicrop results。

關於性能表的標註:

  • ConvNet:卷積神經網路的名稱
  • ImageNet top1 acc:論文中基於 ImageNet 數據集最好的 top1 準確率
  • ImageNet top5 acc:論文中基於 ImageNet 數據集最好的 top5 準確率
  • Published In:論文發表在哪個會議或期刊

論文&代碼

1. VGG

Very Deep Convolutional Networks for Large-Scale Image Recognition.

Karen Simonyan, Andrew Zisserman

pdf: arxiv.org/abs/1409.1556

code: torchvision :

github.com/pytorch/visi

code: keras-applications :

github.com/keras-team/k

code: keras-applications :

github.com/keras-team/k

2. GoogleNet

Going Deeper with Convolutions

Christian Szegedy, Wei Liu, Yangqing Jia, Pierre Sermanet, Scott Reed, Dragomir Anguelov, Dumitru Erhan, Vincent Vanhoucke, Andrew Rabinovich

pdf: arxiv.org/abs/1409.4842

code: unofficial-tensorflow :

github.com/conan7882/Go

code: unofficial-caffe :

github.com/lim0606/caff

3. PReLU-nets

Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification

Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun

pdf: arxiv.org/abs/1502.0185

code: unofficial-chainer :

github.com/nutszebra/pr

4. ResNet

Deep Residual Learning for Image Recognition

Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun

pdf: arxiv.org/abs/1512.0338

code: facebook-torch :

github.com/facebook/fb.

code: torchvision :

github.com/pytorch/visi

code: keras-applications :

github.com/keras-team/k

code: unofficial-keras :

github.com/raghakot/ker

code: unofficial-tensorflow :

github.com/ry/tensorflo

5. PreActResNet

Identity Mappings in Deep Residual Networks

Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun

pdf: arxiv.org/abs/1603.0502

code: facebook-torch :

github.com/facebook/fb.

code: official :

github.com/KaimingHe/re

code: unoffical-pytorch :

github.com/kuangliu/pyt

code: unoffical-mxnet :

github.com/tornadomeet/

6. Inceptionv3

Rethinking the Inception Architecture for Computer Vision

Christian Szegedy, Vincent Vanhoucke, Sergey Ioffe, Jonathon Shlens, Zbigniew Wojna

pdf: arxiv.org/abs/1512.0056

code: torchvision :

github.com/pytorch/visi

code: keras-applications :

github.com/keras-team/k

7. Inceptionv4 && Inception-ResNetv2

Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning

Christian Szegedy, Sergey Ioffe, Vincent Vanhoucke, Alex Alemi

pdf: arxiv.org/abs/1602.0726

code: unofficial-keras :

github.com/kentsommer/k

code: unofficial-keras :

github.com/titu1994/Inc

code: unofficial-keras :

github.com/yuyang-huang

8. RIR

Resnet in Resnet: Generalizing Residual Architectures

Sasha Targ, Diogo Almeida, Kevin Lyman

pdf: arxiv.org/abs/1603.0802

code: unofficial-tensorflow :

github.com/SunnerLi/RiR

code: unofficial-chainer :

github.com/nutszebra/re

9. Stochastic Depth ResNet

Deep Networks with Stochastic Depth

Gao Huang, Yu Sun, Zhuang Liu, Daniel Sedra, Kilian Weinberger

pdf: arxiv.org/abs/1603.0938

code: unofficial-torch :

github.com/yueatsprogra

code: unofficial-chainer :

github.com/yasunorikudo

code: unofficial-keras :

github.com/dblN/stochas

10. WRN

Wide Residual Networks

Sergey Zagoruyko, Nikos Komodakis

pdf: arxiv.org/abs/1605.0714

code: official :

github.com/szagoruyko/w

code: unofficial-pytorch :

github.com/xternalz/Wid

code: unofficial-keras :

github.com/asmith26/wid

code: unofficial-pytorch :

github.com/meliketoy/wi

11. squeezenet

SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size

Forrest N. Iandola, Song Han, Matthew W. Moskewicz, Khalid Ashraf, William J. Dally, Kurt Keutzer

pdf: arxiv.org/abs/1602.0736

code: torchvision :

github.com/pytorch/visi

code: unofficial-caffe :

github.com/DeepScale/Sq

code: unofficial-keras :

github.com/rcmalli/kera

code: unofficial-caffe :

github.com/songhan/Sque

12. GeNet

Genetic CNN

Lingxi Xie, Alan Yuille

pdf: arxiv.org/abs/1703.0151

code: unofficial-tensorflow :

github.com/aqibsaeed/Ge

12. MetaQNN

Designing Neural Network Architectures using Reinforcement Learning

Bowen Baker, Otkrist Gupta, Nikhil Naik, Ramesh Raskar

pdf: arxiv.org/abs/1703.0151

code: official : github.com/bowenbaker/m

13. PyramidNet

Deep Pyramidal Residual Networks

Dongyoon Han, Jiwhan Kim, Junmo Kim

pdf: arxiv.org/abs/1610.0291

code: official :

github.com/jhkim89/Pyra

code: unofficial-pytorch :

github.com/dyhan0920/Py

14. DenseNet

Densely Connected Convolutional Networks

Gao Huang, Zhuang Liu, Laurens van der Maaten, Kilian Q. Weinberger

pdf: arxiv.org/abs/1608.0699

code: official :

github.com/liuzhuang13/

code: unofficial-keras :

github.com/titu1994/Den

code: unofficial-caffe :

github.com/shicai/Dense

code: unofficial-tensorflow :

github.com/YixuanLi/den

code: unofficial-pytorch :

github.com/YixuanLi/den

code: unofficial-pytorch :

github.com/bamos/densen

code: unofficial-keras :

github.com/flyyufelix/D

15. FractalNet

FractalNet: Ultra-Deep Neural Networks without Residuals

Gustav Larsson, Michael Maire, Gregory Shakhnarovich

pdf: arxiv.org/abs/1605.0764

code: unofficial-caffe :

github.com/gustavla/fra

code: unofficial-keras :

github.com/snf/keras-fr

code: unofficial-tensorflow :

github.com/tensorpro/Fr

16. ResNext

Aggregated Residual Transformations for Deep Neural Networks

Saining Xie, Ross Girshick, Piotr Dollár, Zhuowen Tu, Kaiming He

pdf: arxiv.org/abs/1611.0543

code: official :

github.com/facebookrese

code: keras-applications :

github.com/keras-team/k

code: unofficial-pytorch :

github.com/prlz77/ResNe

code: unofficial-keras :

github.com/titu1994/Ker

code: unofficial-tensorflow :

github.com/taki0112/Res

code: unofficial-tensorflow :

github.com/wenxinxu/Res

17. IGCV1

Interleaved Group Convolutions for Deep Neural Networks

Ting Zhang, Guo-Jun Qi, Bin Xiao, Jingdong Wang

pdf: arxiv.org/abs/1707.0272

code official :

github.com/hellozting/I

18. Residual Attention Network

Residual Attention Network for Image Classification

Fei Wang, Mengqing Jiang, Chen Qian, Shuo Yang, Cheng Li, Honggang Zhang, Xiaogang Wang, Xiaoou Tang

pdf: arxiv.org/abs/1704.0690

code: official :

github.com/fwang91/resi

code: unofficial-pytorch :

github.com/tengshaofeng

code: unofficial-gluon :

github.com/PistonY/Resi

code: unofficial-keras :

github.com/koichiro11/r

19. Xception

Xception: Deep Learning with Depthwise Separable Convolutions

Fran?ois Chollet

pdf: arxiv.org/abs/1610.0235

code: unofficial-pytorch :

github.com/jfzhang95/py

code: unofficial-tensorflow :

github.com/kwotsin/Tens

code: unofficial-caffe :

github.com/yihui-he/Xce

code: unofficial-pytorch :

github.com/tstandley/Xc

code: keras-applications :

github.com/keras-team/k

20. MobileNet

MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications

Andrew G. Howard, Menglong Zhu, Bo Chen, Dmitry Kalenichenko, Weijun Wang, Tobias Weyand, Marco Andreetto, Hartwig Adam

pdf: arxiv.org/abs/1704.0486

code: unofficial-tensorflow :

github.com/Zehaos/Mobil

code: unofficial-caffe :

github.com/shicai/Mobil

code: unofficial-pytorch :

github.com/marvis/pytor

code: keras-applications :

github.com/keras-team/k

21. PolyNet

PolyNet: A Pursuit of Structural Diversity in Very Deep Networks

Xingcheng Zhang, Zhizhong Li, Chen Change Loy, Dahua Lin

pdf: arxiv.org/abs/1611.0572

code: official :

github.com/open-mmlab/p

22. DPN

Dual Path Networks

Yunpeng Chen, Jianan Li, Huaxin Xiao, Xiaojie Jin, Shuicheng Yan, Jiashi Feng

pdf: arxiv.org/abs/1707.0162

code: official :

github.com/cypw/DPNs

code: unoffical-keras :

github.com/titu1994/Ker

code: unofficial-pytorch :

github.com/oyam/pytorch

code: unofficial-pytorch :

github.com/rwightman/py

23. Block-QNN

Practical Block-wise Neural Network Architecture Generation

Zhao Zhong, Junjie Yan, Wei Wu, Jing Shao, Cheng-Lin Liu

pdf: arxiv.org/abs/1708.0555

24. CRU-Net

Sharing Residual Units Through Collective Tensor Factorization in Deep Neural Networks

Chen Yunpeng, Jin Xiaojie, Kang Bingyi, Feng Jiashi, Yan Shuicheng

pdf: arxiv.org/abs/1703.0218

code official :

github.com/cypw/CRU-Net

code unofficial-mxnet :

github.com/bruinxiong/M

25. ShuffleNet

ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices

Xiangyu Zhang, Xinyu Zhou, Mengxiao Lin, Jian Sun

pdf: arxiv.org/abs/1707.0108

code: unofficial-tensorflow :

github.com/MG2033/Shuff

code: unofficial-pytorch :

github.com/jaxony/Shuff

code: unofficial-caffe :

github.com/farmingyard/

code: unofficial-keras :

github.com/scheckmedia/

26. CondenseNet

CondenseNet: An Efficient DenseNet using Learned Group Convolutions

Gao Huang, Shichen Liu, Laurens van der Maaten, Kilian Q. Weinberger

pdf: arxiv.org/abs/1711.0922

code: official :

github.com/ShichenLiu/C

code: unofficial-tensorflow :

github.com/markdtw/cond

27. NasNet

Learning Transferable Architectures for Scalable Image Recognition

Barret Zoph, Vijay Vasudevan, Jonathon Shlens, Quoc V. Le

pdf: arxiv.org/abs/1707.0701

code: unofficial-keras :

github.com/titu1994/Ker

code: keras-applications :

github.com/keras-team/k

code: unofficial-pytorch :

github.com/wandering007

code: unofficial-tensorflow :

github.com/yeephycho/na

28. MobileNetV2

MobileNetV2: Inverted Residuals and Linear Bottlenecks

Mark Sandler, Andrew Howard, Menglong Zhu, Andrey Zhmoginov, Liang-Chieh Chen

pdf: arxiv.org/abs/1801.0438

code: unofficial-keras :

github.com/xiaochus/Mob

code: unofficial-pytorch :

github.com/Randl/Mobile

code: unofficial-tensorflow :

github.com/neuleaf/Mobi

29. IGCV2

IGCV2: Interleaved Structured Sparse Convolutional Neural Networks

Guotian Xie, Jingdong Wang, Ting Zhang, Jianhuang Lai, Richang Hong, Guo-Jun Qi

pdf: arxiv.org/abs/1804.0620

30. hier

Hierarchical Representations for Efficient Architecture Search

Hanxiao Liu, Karen Simonyan, Oriol Vinyals, Chrisantha Fernando, Koray Kavukcuoglu

pdf: arxiv.org/abs/1711.0043

31. PNasNet

Progressive Neural Architecture Search

Chenxi Liu, Barret Zoph, Maxim Neumann, Jonathon Shlens, Wei Hua, Li-Jia Li, Li Fei-Fei, Alan Yuille, Jonathan Huang, Kevin Murphy

pdf: arxiv.org/abs/1712.0055

code: tensorflow-slim :

github.com/tensorflow/m

code: unofficial-pytorch :

github.com/chenxi116/PN

code: unofficial-tensorflow :

github.com/chenxi116/PN

32. AmoebaNet

Regularized Evolution for Image Classifier Architecture Search

Esteban Real, Alok Aggarwal, Yanping Huang, Quoc V Le

pdf: arxiv.org/abs/1802.0154

code: tensorflow-tpu :

github.com/tensorflow/t

33. SENet

Squeeze-and-Excitation Networks

Jie Hu, Li Shen, Samuel Albanie, Gang Sun, Enhua Wu

pdf: arxiv.org/abs/1709.0150

code: official :

github.com/hujie-frank/

code: unofficial-pytorch :

github.com/moskomule/se

code: unofficial-tensorflow :

github.com/taki0112/SEN

code: unofficial-caffe :

github.com/shicai/SENet

code: unofficial-mxnet :

github.com/bruinxiong/S

34. ShuffleNetV2

ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design

Ningning Ma, Xiangyu Zhang, Hai-Tao Zheng, Jian Sun

pdf: arxiv.org/abs/1807.1116

code: unofficial-pytorch :

github.com/Randl/Shuffl

code: unofficial-keras :

github.com/opconty/kera

code: unofficial-pytorch :

github.com/Bugdragon/Sh

code: unofficial-caff2:

github.com/wolegechu/Sh

35. IGCV3

IGCV3: Interleaved Low-Rank Group Convolutions for Efficient Deep Neural Networks

Ke Sun, Mingjie Li, Dong Liu, Jingdong Wang

pdf: arxiv.org/abs/1806.0017

code: official :

github.com/homles11/IGC

code: unofficial-pytorch :

github.com/xxradon/IGCV

code: unofficial-tensorflow :

github.com/ZHANG-SHI-CH

36. MNasNet

MnasNet: Platform-Aware Neural Architecture Search for Mobile

Mingxing Tan, Bo Chen, Ruoming Pang, Vijay Vasudevan, Quoc V. Le

pdf: arxiv.org/abs/1807.1162

code: unofficial-pytorch :

github.com/AnjieZheng/M

code: unofficial-caffe :

github.com/LiJianfei06/

code: unofficial-MxNet :

github.com/chinakook/Mn

code: unofficial-keras :

github.com/Shathe/MNasN

更多 AI 乾貨,請關注公眾號:AI有道(ID:redstonewill)


推薦閱讀:
相關文章