1.去掉Data层中的transform_param中的mean_file和backend,会自动读取之前数据集里生成的,mirror和crop_size不受影响,需要就填,不需要就不填
2.改 include {
phase: TRAIN
}
为 include { stage: "train" } ,同理TEST改成val,其实这个不改好像也可以,具体可以查看自定义网络旁边的问号帮助
3.修改最后一个全连接层的numoutput(这个其实不算吧,具体根据自己的填,只是提醒下别忘了)
4.最后一个全连接层后的网络替换为(ip2为最后一个全连接层),其实就是把之前的网络里的train_val和deploy给合并了
layer {
name: "accuracy"
type: "Accuracy"
bottom: "ip2"
bottom: "label"
top: "accuracy"
include { stage: "val" }
}
layer {
name: "loss"
type: "SoftmaxWithLoss"
bottom: "ip2"
bottom: "label"
top: "loss"
exclude { stage: "deploy" }
}
layer {
name: "softmax"
type: "Softmax"
bottom: "ip2"
top: "softmax"
include { stage: "deploy" }
2.改 include {
phase: TRAIN
}
为 include { stage: "train" } ,同理TEST改成val,其实这个不改好像也可以,具体可以查看自定义网络旁边的问号帮助
3.修改最后一个全连接层的numoutput(这个其实不算吧,具体根据自己的填,只是提醒下别忘了)
4.最后一个全连接层后的网络替换为(ip2为最后一个全连接层),其实就是把之前的网络里的train_val和deploy给合并了
layer {
name: "accuracy"
type: "Accuracy"
bottom: "ip2"
bottom: "label"
top: "accuracy"
include { stage: "val" }
}
layer {
name: "loss"
type: "SoftmaxWithLoss"
bottom: "ip2"
bottom: "label"
top: "loss"
exclude { stage: "deploy" }
}
layer {
name: "softmax"
type: "Softmax"
bottom: "ip2"
top: "softmax"
include { stage: "deploy" }
}