从GoogleNet中提取分数

时间:2017-04-11 11:49:03

标签: caffe

我想从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

有什么建议吗?

1 个答案:

答案 0 :(得分:0)

图层loss3/top5Accuracy 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部分。