deploy.prototxt
1.删除输入数据(如:type:data...inckude{phase: TRAIN}),然后添加一个数据维度描述。
input: "data"
input_dim: 1
input_dim: 3
input_dim: 224
input_dim: 224
force_backward: true
2.移除最后的“loss” 和“accuracy” 层,加入“prob”层。
layers {
name: "prob"
type: SOFTMAX
bottom: "fc8" //只有一个bottom,label移除
top: "prob"
}
转载地址: http://blog.csdn.net/wang4959520/article/details/51841110