首页 文章

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

提问于
浏览 302
2

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

非常感谢

1 回答

  • 0

    正如您在infer.py中的代码中所看到的,预测标签是 argmax'score' 层 .

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

    'score' 是训练期间的input to "SoftmaxWithLoss" layer . 因此,要从 'score' 获取类概率,您需要在 'score' 之上添加 "Softmax"

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

相关问题