[转]caffe-ssd中 net.forward()返回值

 

out = net.forward()

在运行caffe的模型中,经常会使用句代码,这个代码中out里面的内容是什么呢?
type(out)可以发现out是一个dict
执行下面的代码:
for key in out:
print key
发现out只有一个key:detection_out,这个是我的caffe输出层的名字,所以out其实就是caffe运行输出的结果。
[转]caffe-ssd中 net.forward()返回值_第1张图片
那么这个结果的具体内容又是什么呢?
在spyder上的运行后显示如下:
[转]caffe-ssd中 net.forward()返回值_第2张图片

bbb = out[‘detection_out’]
bbb[0,0,:,1:7]
在最后一维,第二个开始依次是:标签、置信度、目标位置的4个坐标信息[xmin ymin xmax ymax]
倒数第二维是识别的标签的数量,图中选择显示第4个标签的信息
[转]caffe-ssd中 net.forward()返回值_第3张图片

 

你可能感兴趣的:(技术_机器学习)