Toybrick

楼主: ddcat1991

onnx模型转换为rknn模型, 输出结果不正确

sotnly

新手上路

积分
35
楼主
发表于 2020-6-5 14:26:14 | 显示全部楼层
大佬,我也刚好在哢这个模型,有好些问题,有偿求助,可以的话加我QQ:405631964,谢谢了
回复

使用道具 举报

sotnly

新手上路

积分
35
沙发
发表于 2020-6-5 14:42:15 | 显示全部楼层
我也是用Ultra-Light-Fast-Generic-Face-Detector-1MB,这个的RFB-640的模型,我做了相应的裁剪,然后也用onnxsim做了模型简化,但是当我把简化的模型转为rknn的时候就报错了:

IndexError:_Map_base::at
我百度了,说是我的torch.onnx.export需要个  keep_initializers_as_inputs 这个参数,我现在的版本根本就没有这个参数,我的就是pytorch1.2.0直接转的onnx1.4.1的,所以不知道该怎么继续,如果你也有这个经历,麻烦告知,谢谢
回复

使用道具 举报

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

本版积分规则

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


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