2017-04-11 19 views
-1

我想從caffe模型中提取GoogleNet網絡的分數,但我不太明白哪些分數保留分數。從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

這看起來像你試圖從訓練有素的模型中獲得前5的準確性,而不是得分本身。你能更具體地說明你需要的「分數」的含義嗎? – Prune

回答

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