Toybrick

caffe获取 两个输出

troy

版主

积分
2352
楼主
发表于 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。
回复

使用道具 举报

troy

版主

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

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

使用道具 举报

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

本版积分规则

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


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