我想从caffe模型中提取GoogleNet网络中的分数,但我不太清楚哪些图层会保留分数。
我明白了:
Check failed: feature_extraction_net->has_blob(blob_names[i]) Unknown feature blob name loss3/top-5 in the network ./train_val.prototxt
有什么建议吗?
答案 0 :(得分:0)
图层loss3/top5
是Accuracy layer。我会假设你希望你的意思是得分。
默认情况下,你会有这样的东西
layer {
name: "loss3/top-5"
type: "Accuracy"
bottom: "loss3/classifier"
bottom: "label"
top: "loss3/top-5"
include {
phase: TEST
}
accuracy_param {
top_k: 5
}
}
如果您希望在培训中使用此图层,则可以删除或注释掉#
部分include
部分。