Neural network 如何在caffe中保存预测的概率？

有人知道如何在FCN语义分割中保存每个像素的预测类标签的输出吗？我想在推断过程中看到图像的概率图。数据应保存在哪一层

---

非常感谢

正如您可以从预测标签中的代码中看到的那样，

'score'

层的

argmax

out = net.blobs['score'].data[0].argmax(axis=0)

“分数”

是培训过程中最重要的分数。因此，要从

'score'

中获取类概率，您需要在

'score'

的顶部添加

“Softmax”

：

e_s = np.exp(net.blobs['score'].data[0])
prob = e_s / e_s.sum(axis=0) 

---

谢谢。。我能看到可能性；但是，对于每个像素，所有这些都是相同的。输出全为零（黑色图像）。您可能需要查看网络，看看哪里出了问题。