Toybrick

Caffe mobilenet-ssd 不支持转rknn

RK3399

注册会员

积分
59
楼主
发表于 2019-4-17 09:06:45 | 显示全部楼层
我也是,目前测试来看并不支持detectoutput层,但是删除该层仍然有问题,模型转换默认会删除prior box层,输出只有两个loc和conf,看他的例子应该是prior box需要用caffe预先运行出来,然后结合loc和conf自己去算。
回复

使用道具 举报

RK3399

注册会员

积分
59
沙发
发表于 2019-4-17 09:10:40 | 显示全部楼层
回复

使用道具 举报

RK3399

注册会员

积分
59
板凳
发表于 2019-4-18 10:48:26 | 显示全部楼层
我解决了,哈哈,坑太多了
RK3399 Pro 的确不支持detectoutput层,需要删除,模型转换默认也会删除prior box层,输出只有两个loc和conf,prior box需要需要先用caffe预先运行出来,然后结合loc和conf自己去算。
回复

使用道具 举报

RK3399

注册会员

积分
59
地板
发表于 2019-4-18 17:04:41 | 显示全部楼层
有空再写
回复

使用道具 举报

RK3399

注册会员

积分
59
5#
发表于 2019-4-19 14:31:55 | 显示全部楼层
771488533 发表于 2019-4-19 13:52
把 detectoutput层 删除了那还要这模型干嘛。。。

detectoutput 删除了,但是可以根据loc, conf , priorbox的运行结果,自己写detectoutput 然后输出正确检测
回复

使用道具 举报

RK3399

注册会员

积分
59
6#
发表于 2019-4-19 15:16:17 | 显示全部楼层
q5671229 发表于 2019-4-19 11:42
写个教程吧。。git上贡献一下源码?   搞不定

可以关注我的仓库
https://github.com/Pinnh/NPU_CaffeSSD
回复

使用道具 举报

RK3399

注册会员

积分
59
7#
发表于 2019-4-19 18:21:53 | 显示全部楼层

记得star哦
回复

使用道具 举报

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

本版积分规则

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


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