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

Question

有谁知道如何在FCN语义分割中保存每个像素的预测类标签的输出？我想在推理期间看到图像的概率图。应该保存图层的数据？

非常感谢

Answer 1

正如您从infer.py中的代码中看到的那样，预测标签为argmax图层的'score'。

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

培训期间'score'是input to "SoftmaxWithLoss" layer。因此，要从'score'获取课程概率，您需要在"Softmax"之上添加'score'：

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