PROSAGA码农传奇-Caffe-Caffe：打印softmax分数

Caffe：打印softmax分数

作者: 筱梨
发布时间: 2024-12-18 12:57:09 (9天前)
转自：

<div class =“post-text”itemprop =“text”>
  
    我可以尝试回答你的问题。假设在部署网络时，softmax层如下所示：
  
   <pre>
 <code>
 layer {
 name: "prob"
 type : "Softmax"
 bottom: "fc6"
 top: "prob"
}

</code>
 </pre>
  
    在处理数据的python代码中，结合@Shai提供的代码，您可以通过添加基于@ Shai代码的代码来获取每个类别的概率：
  
   <pre>
 <code>
 predicted_prob = net.blobs['prob'].data

</code>
 </pre>
  
    predict_prob将返回一个包含所有类别概率的数组。
  
  
    例如，如果您只有两个类别，
     <code>
 predicted_prob[0][0]
 </code>
     将是此测试数据属于一个类别的概率
     <code>
 predicted_prob[0][1]
 </code>
     将是另一个的概率。
  
  
    PS：
  
  
    如果你不想写任何额外的python脚本，根据
    <a href="https://github.com/BVLC/caffe/tree/master/examples/mnist" rel="nofollow noreferrer">
      https://github.com/BVLC/caffe/tree/master/examples/mnist
    </A>
     
它说这个例子将自动每500次迭代进行一次测试。 “500”在求解器中定义，例如
    <a href="https://github.com/BVLC/caffe/blob/master/examples/mnist/lenet_solver.prototxt" rel="nofollow noreferrer">
      https://github.com/BVLC/caffe/blob/master/examples/mnist/lenet_solver.prototxt
    </A>
  
  
    因此，您需要追溯处理解算器文件的caffe源代码。我想它应该是
    <a href="https://github.com/BVLC/caffe/blob/master/src/caffe/solver.cpp" rel="nofollow noreferrer">
      https://github.com/BVLC/caffe/blob/master/src/caffe/solver.cpp
    </A>
  
  
    我不确定solver.cpp是你需要查看的正确文件。但是在这个文件中，你可以看到它具有测试和计算某些值的功能。如果没有其他人可以回答你的问题，我希望它可以给你一些想法。
  
</DIV>