Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- name: "NN2"
- layer {
- name: "data"
- type: "Data"
- top: "data"
- top: "label"
- include {
- phase: TRAIN
- }
- transform_param {
- mirror: true
- mean_file :"/home/jiayi-wei/caffe/examples/NN2/image_train_mean.binaryproto"
- data_param {
- source: "/home/jiayi-wei/caffe/examples/NN2/img_train_lmdb"
- batch_size: 30
- backend: LMDB
- }
- }
- layer {
- name: "data"
- type: "Data"
- top: "data"
- top: "label"
- include {
- phase: TEST
- }
- transform_param {
- mirror: false
- mean_file :"/home/jiayi-wei/caffe/examples/NN2/image_train_mean.binaryproto"
- data_param {
- source: "/home/jiayi-wei/caffe/examples/NN2/img_val_lmdb"
- batch_size: 11
- backend: LMDB
- }
- }
- #first layers
- layer {
- name: "conv11"
- type: "Convolution"
- bottom: "data"
- top: "conv11"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 64
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu11"
- type: "ReLU"
- bottom: "conv11"
- top: "conv11"
- }
- layer {
- name: "conv12"
- type: "Convolution"
- bottom: "conv11"
- top: "conv12"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 128
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu12"
- type: "ReLU"
- bottom: "conv12"
- top: "conv12"
- }
- layer {
- name: "pool1"
- type: "Pooling"
- bottom: "conv12"
- top: "pool1"
- pooling_param {
- pool: MAX
- kernel_size: 2
- stride: 2
- }
- }
- #second layers
- layer {
- name: "conv21"
- type: "Convolution"
- bottom: "pool1"
- top: "conv21"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 64
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu21"
- type: "ReLU"
- bottom: "conv21"
- top: "conv21"
- }
- layer {
- name: "conv22"
- type: "Convolution"
- bottom: "conv21"
- top: "conv22"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 128
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu22"
- type: "ReLU"
- bottom: "conv22"
- top: "conv22"
- }
- layer {
- name: "pool2"
- type: "Pooling"
- bottom: "conv22"
- top: "pool2"
- pooling_param {
- pool: MAX
- kernel_size: 2
- stride: 2
- }
- }
- #third layers
- layer {
- name: "conv31"
- type: "Convolution"
- bottom: "pool2"
- top: "conv31"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 128
- pad:1
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu31"
- type: "ReLU"
- bottom: "conv31"
- top: "conv31"
- }
- layer {
- name: "conv32"
- type: "Convolution"
- bottom: "conv31"
- top: "conv32"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 128
- pad:1
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu32"
- type: "ReLU"
- bottom: "conv32"
- top: "conv32"
- }
- layer {
- name: "pool3"
- type: "Pooling"
- bottom: "conv32"
- top: "pool3"
- pooling_param {
- pool: MAX
- pad:1
- kernel_size: 2
- stride: 2
- }
- }
- #fourth layer
- layer {
- name: "conv41"
- type: "Convolution"
- bottom: "pool3"
- top: "conv41"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 256
- pad:1
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu41"
- type: "ReLU"
- bottom: "conv41"
- top: "conv41"
- }
- layer {
- name: "conv42"
- type: "Convolution"
- bottom: "conv41"
- top: "conv42"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 256
- pad:1
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu42"
- type: "ReLU"
- bottom: "conv42"
- top: "conv42"
- }
- layer {
- name: "conv43"
- type: "Convolution"
- bottom: "conv42"
- top: "conv43"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 256
- pad:1
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu43"
- type: "ReLU"
- bottom: "conv43"
- top: "conv43"
- }
- layer {
- name: "pool4"
- type: "Pooling"
- bottom: "conv43"
- top: "pool4"
- pooling_param {
- pool: MAX
- kernel_size: 2
- stride: 2
- }
- }
- #fiveth layer
- layer {
- name: "conv51"
- type: "Convolution"
- bottom: "pool4"
- top: "conv51"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 256
- pad:1
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu51"
- type: "ReLU"
- bottom: "conv51"
- top: "conv51"
- }
- layer {
- name: "conv52"
- type: "Convolution"
- bottom: "conv51"
- top: "conv52"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 256
- pad:1
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "relu52"
- type: "ReLU"
- bottom: "conv52"
- top: "conv52"
- }
- layer {
- name: "conv53"
- type: "Convolution"
- bottom: "conv52"
- top: "conv53"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- convolution_param {
- num_output: 256
- pad:1
- kernel_size: 3
- stride: 1
- weight_filler {
- type: "gaussian"
- std: 0.01
- }
- bias_filler {
- type: "constant"
- value: 0
- }
- }
- }
- layer {
- name: "pool5"
- type: "Pooling"
- bottom: "conv53"
- top: "pool5"
- pooling_param {
- pool: AVE
- pad:1
- kernel_size: 2
- stride: 2
- }
- }
- #drop_Fc
- layer {
- name: "dropout"
- type: "Dropout"
- bottom: "pool5"
- top: "pool5"
- dropout_param {
- dropout_ratio: 0.5
- }
- }
- layer {
- name: "fc6"
- type: "InnerProduct"
- bottom: "pool5"
- top: "fc6"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- inner_product_param {
- num_output:1000
- weight_filler {
- type: "gaussian"
- std: 0.005
- }
- bias_filler {
- type: "constant"
- value: 1
- }
- }
- }
- layer {
- name: "fc7"
- type: "InnerProduct"
- bottom: "fc6"
- top: "fc7"
- param {
- lr_mult: 1
- decay_mult: 1
- }
- param {
- lr_mult: 2
- decay_mult: 0
- }
- inner_product_param {
- num_output:10575
- weight_filler {
- type: "gaussian"
- std: 0.005
- }
- bias_filler {
- type: "constant"
- value: 1
- }
- }
- }
- layer {
- name: "accuracy"
- type: "Accuracy"
- bottom: "fc7"
- bottom: "label"
- top: "accuracy"
- include {
- phase: TEST
- }
- }
- layer {
- name: "SoftMax"
- type: "SoftmaxWithLoss"
- bottom: "fc7"
- bottom: "label"
- top: "SoftMax"
- }
- net: "train_val.prototxt"
- test_iter: 10000
- test_interval: 1000
- base_lr: 0.01
- lr_policy: "step"
- gamma: 0.1
- stepsize: 100000
- display: 20
- max_iter: 450000
- momentum: 0.9
- weight_decay: 0.0005
- snapshot: 10000
- snapshot_prefix: "/home/jiayi-wei/caffe/examples/NN2"
- solver_mode: GPU
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement