summaryrefslogtreecommitdiff
path: root/examples/imagenet/alexnet_val.prototxt
diff options
context:
space:
mode:
Diffstat (limited to 'examples/imagenet/alexnet_val.prototxt')
-rw-r--r--examples/imagenet/alexnet_val.prototxt180
1 files changed, 81 insertions, 99 deletions
diff --git a/examples/imagenet/alexnet_val.prototxt b/examples/imagenet/alexnet_val.prototxt
index aff33d01..3fd6296e 100644
--- a/examples/imagenet/alexnet_val.prototxt
+++ b/examples/imagenet/alexnet_val.prototxt
@@ -1,40 +1,38 @@
name: "AlexNet"
layers {
- layer {
- name: "data"
- type: "data"
+ name: "data"
+ type: DATA
+ data_param {
source: "ilsvrc12_val_leveldb"
- meanfile: "../../data/ilsvrc12/imagenet_mean.binaryproto"
- batchsize: 50
- cropsize: 227
+ mean_file: "../../data/ilsvrc12/imagenet_mean.binaryproto"
+ batch_size: 50
+ crop_size: 227
mirror: false
}
top: "data"
top: "label"
}
layers {
- layer {
- name: "conv1"
- type: "conv"
+ name: "conv1"
+ type: CONVOLUTION
+ convolution_param {
num_output: 96
- kernelsize: 11
+ kernel_size: 11
stride: 4
}
bottom: "data"
top: "conv1"
}
layers {
- layer {
- name: "relu1"
- type: "relu"
- }
+ name: "relu1"
+ type: RELU
bottom: "conv1"
top: "conv1"
}
layers {
- layer {
- name: "norm1"
- type: "lrn"
+ name: "norm1"
+ type: LRN
+ lrn_param {
local_size: 5
alpha: 0.0001
beta: 0.75
@@ -43,40 +41,38 @@ layers {
top: "norm1"
}
layers {
- layer {
- name: "pool1"
- type: "pool"
+ name: "pool1"
+ type: POOLING
+ pooling_param {
pool: MAX
- kernelsize: 3
+ kernel_size: 3
stride: 2
}
bottom: "norm1"
top: "pool1"
}
layers {
- layer {
- name: "conv2"
- type: "conv"
+ name: "conv2"
+ type: CONVOLUTION
+ convolution_param {
num_output: 256
- group: 2
- kernelsize: 5
pad: 2
+ kernel_size: 5
+ group: 2
}
bottom: "pool1"
top: "conv2"
}
layers {
- layer {
- name: "relu2"
- type: "relu"
- }
+ name: "relu2"
+ type: RELU
bottom: "conv2"
top: "conv2"
}
layers {
- layer {
- name: "norm2"
- type: "lrn"
+ name: "norm2"
+ type: LRN
+ lrn_param {
local_size: 5
alpha: 0.0001
beta: 0.75
@@ -85,161 +81,147 @@ layers {
top: "norm2"
}
layers {
- layer {
- name: "pool2"
- type: "pool"
+ name: "pool2"
+ type: POOLING
+ pooling_param {
pool: MAX
- kernelsize: 3
+ kernel_size: 3
stride: 2
}
bottom: "norm2"
top: "pool2"
}
layers {
- layer {
- name: "conv3"
- type: "conv"
+ name: "conv3"
+ type: CONVOLUTION
+ convolution_param {
num_output: 384
- kernelsize: 3
pad: 1
+ kernel_size: 3
}
bottom: "pool2"
top: "conv3"
}
layers {
- layer {
- name: "relu3"
- type: "relu"
- }
+ name: "relu3"
+ type: RELU
bottom: "conv3"
top: "conv3"
}
layers {
- layer {
- name: "conv4"
- type: "conv"
+ name: "conv4"
+ type: CONVOLUTION
+ convolution_param {
num_output: 384
- group: 2
- kernelsize: 3
pad: 1
+ kernel_size: 3
+ group: 2
}
bottom: "conv3"
top: "conv4"
}
layers {
- layer {
- name: "relu4"
- type: "relu"
- }
+ name: "relu4"
+ type: RELU
bottom: "conv4"
top: "conv4"
}
layers {
- layer {
- name: "conv5"
- type: "conv"
+ name: "conv5"
+ type: CONVOLUTION
+ convolution_param {
num_output: 256
- group: 2
- kernelsize: 3
pad: 1
+ kernel_size: 3
+ group: 2
}
bottom: "conv4"
top: "conv5"
}
layers {
- layer {
- name: "relu5"
- type: "relu"
- }
+ name: "relu5"
+ type: RELU
bottom: "conv5"
top: "conv5"
}
layers {
- layer {
- name: "pool5"
- type: "pool"
- kernelsize: 3
+ name: "pool5"
+ type: POOLING
+ pooling_param {
pool: MAX
+ kernel_size: 3
stride: 2
}
bottom: "conv5"
top: "pool5"
}
layers {
- layer {
- name: "fc6"
- type: "innerproduct"
+ name: "fc6"
+ type: INNER_PRODUCT
+ inner_product_param {
num_output: 4096
}
bottom: "pool5"
top: "fc6"
}
layers {
- layer {
- name: "relu6"
- type: "relu"
- }
+ name: "relu6"
+ type: RELU
bottom: "fc6"
top: "fc6"
}
layers {
- layer {
- name: "drop6"
- type: "dropout"
+ name: "drop6"
+ type: DROPOUT
+ dropout_param {
dropout_ratio: 0.5
}
bottom: "fc6"
top: "fc6"
}
layers {
- layer {
- name: "fc7"
- type: "innerproduct"
+ name: "fc7"
+ type: INNER_PRODUCT
+ inner_product_param {
num_output: 4096
}
bottom: "fc6"
top: "fc7"
}
layers {
- layer {
- name: "relu7"
- type: "relu"
- }
+ name: "relu7"
+ type: RELU
bottom: "fc7"
top: "fc7"
}
layers {
- layer {
- name: "drop7"
- type: "dropout"
+ name: "drop7"
+ type: DROPOUT
+ dropout_param {
dropout_ratio: 0.5
}
bottom: "fc7"
top: "fc7"
}
layers {
- layer {
- name: "fc8"
- type: "innerproduct"
+ name: "fc8"
+ type: INNER_PRODUCT
+ inner_product_param {
num_output: 1000
}
bottom: "fc7"
top: "fc8"
}
layers {
- layer {
- name: "prob"
- type: "softmax"
- }
+ name: "prob"
+ type: SOFTMAX
bottom: "fc8"
top: "prob"
}
layers {
- layer {
- name: "accuracy"
- type: "accuracy"
- }
+ top: "accuracy"
+ name: "accuracy"
+ type: ACCURACY
bottom: "prob"
bottom: "label"
- top: "accuracy"
}