Toybrick

caffe获取 两个输出

tof3d

中级会员

积分
210
发表于 2019-3-19 17:42:28    查看: 1559|回复: 9 | [复制链接]    打印 | 显示全部楼层
        out = PNet.forward()
        print(out['prob1'])
        pdb.set_trace()
   
        boxes = generateBoundingBox(out['prob1'][0,1,:,:], out['conv4-2'][0], scale, threshold[0])
需要获取prob1 与conv4-2 层的结果,要怎么做呢

回复

使用道具 举报

tof3d

中级会员

积分
210
 楼主| 发表于 2019-3-19 18:52:55 | 显示全部楼层
本帖最后由 tof3d 于 2019-3-19 19:32 编辑

python可以吗,还是要C++,我需要或者prob1 ,以及前面的conv4-2 层
回复

使用道具 举报

tof3d

中级会员

积分
210
 楼主| 发表于 2019-3-19 19:31:11 | 显示全部楼层
为什么python的rknn_outputs_get api 看不到
回复

使用道具 举报

troy

版主

积分
1758
发表于 2019-3-20 08:36:52 | 显示全部楼层
tof3d 发表于 2019-3-19 19:31
为什么python的rknn_outputs_get api 看不到

要获取中间层的结果,转换模型的时候,可以把它加到输出列表里面,然后interface接口就会一起返回了。
要注意的一点是,两个上下层之间不能同时输出,比如A->B->C, 可以指定输出A和C,但是不知指定输出AB或者BC。
回复

使用道具 举报

tof3d

中级会员

积分
210
 楼主| 发表于 2019-3-20 09:23:23 | 显示全部楼层
我用的不是tensorflow啊,caffe转模型可以设置output吗,可以举个例子吗,还有我怀疑caffe的输出,是最后一层,还是最后两层的结果
回复

使用道具 举报

tof3d

中级会员

积分
210
 楼主| 发表于 2019-3-20 09:24:01 | 显示全部楼层
本帖最后由 tof3d 于 2019-3-20 09:39 编辑
troy 发表于 2019-3-20 08:36
要获取中间层的结果,转换模型的时候,可以把它加到输出列表里面,然后interface接口就会一起返回了。
要 ...

我用的不是tensorflow啊,caffe转模型可以设置output吗,可以举个例子吗,我的最后两层是prob1与conv4-2,
也是奇怪为什么输出不可以互相连接的两层,还有我怀疑caffe的最后输出,是最后一层,还是最后两层的结果,怎么感觉最后softmax输出那么多东西呢,我现在在移植mtcnn,你们是不是成功了,可以共享一份代码吗,我邮箱3459794356@qq.com
回复

使用道具 举报

troy

版主

积分
1758
发表于 2019-3-20 10:14:47 | 显示全部楼层
tof3d 发表于 2019-3-20 09:24
我用的不是tensorflow啊,caffe转模型可以设置output吗,可以举个例子吗,我的最后两层是prob ...

caffe的我不太熟悉,应该也是有地方可以指定输出的,有类似的操作
回复

使用道具 举报

tof3d

中级会员

积分
210
 楼主| 发表于 2019-3-20 10:25:55 | 显示全部楼层
troy 发表于 2019-3-20 10:14
caffe的我不太熟悉,应该也是有地方可以指定输出的,有类似的操作

那就说python那边没有暴露这样的接口是吗
回复

使用道具 举报

tof3d

中级会员

积分
210
 楼主| 发表于 2019-3-20 10:38:24 | 显示全部楼层
本帖最后由 tof3d 于 2019-3-20 10:42 编辑
tof3d 发表于 2019-3-20 10:25
那就说python那边没有暴露这样的接口是吗

还有python输入是否修改对应可以matlab的mat格式的输入 类似C++的 inputs[0].fmt = RKNN_TENSOR_NHWC;
回复

使用道具 举报

tof3d

中级会员

积分
210
 楼主| 发表于 2019-3-21 11:27:34 | 显示全部楼层
已解决把,看起来RK确实输出最后两层的结果
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表