从GoogleNet中提取分数

Question

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

有什么建议吗？

Answer 1

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