AlexNet详解

时间:2022-08-24 23:55:10

在imagenet上的图像分类challenge上Alex提出的alexnet网络结构模型赢得了2012届的冠军。要研究CNN类型DL网络模型在图像分类上的应用,就逃不开研究alexnet,这是CNN在图像分类上的经典模型(DL火起来之后)。

在DL开源实现caffe的model样例中,它也给出了alexnet的复现,具体网络配置文件如下https://github.com/BVLC/caffe/blob/master/models/bvlc_reference_caffenet/train_val.prototxt

接下来本文将一步步对该网络配置结构中各个层进行详细的解读(训练阶段):

1. conv1阶段DFD(data flow diagram):

AlexNet详解

2. conv2阶段DFD(data flow diagram):

AlexNet详解

3. conv3阶段DFD(data flow diagram):

AlexNet详解

4. conv4阶段DFD(data flow diagram):

AlexNet详解

5. conv5阶段DFD(data flow diagram):

AlexNet详解

6. fc6阶段DFD(data flow diagram):

AlexNet详解

7. fc7阶段DFD(data flow diagram):

AlexNet详解

8. fc8阶段DFD(data flow diagram):

AlexNet详解

各种layer的operation更多解释可以参考http://caffe.berkeleyvision.org/tutorial/layers.html

从计算该模型的数据流过程中,该模型参数大概5kw+。