OpenCV 中使用Caffe中训练好的model

在OpenCV中使用Caffe中训练好的model

以 LeNet 为例子

第一步： 改写deploy文件

按下面的步骤修改文件 lenet_train_test.prototxt

(1) 去掉数据输入层，即将top 为 “data” 的layers 去掉

      即将下面的内容删掉     

layer {
  name: "mnist"
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
  transform_param {
    scale: 0.00390625
  }
  data_param {
    source: "examples/mnist/mnist_train_lmdb"
    batch_size: 64
    backend: LMDB
  }
}
layer {
  name: "mnist"
  type: "Data"
  top: "data"
  top: "label"
  include {
    phase: TEST
  }
  transform_param {
    scale: 0.00390625
  }
  data_param {
    source: "examples/mnist/mnist_test_lmdb"
    batch_size: 100
    backend: LMDB
  }
}

(2) 重新建立输入

     即添加下面的内容

input: "data"
input_shape {
  dim: 1   # batchsize
  dim: 1   # number of channels 
  dim: 28  # width
  dim: 28  # height
}

(3) 去掉输出层，即将bottom 为 “label” 的layers 去掉

     即将下面的内容删掉