Toybrick

标题: 改成自己的类别后报错维度不匹配 [打印本页]

作者: wang64 时间: 2020-7-3 16:43
标题: 改成自己的类别后报错维度不匹配
用tensorflow mobilenetssd 训练自己的类别后得到的pb模型可以正常检测，转换成rknn模型后在pc端上也可以正常检测。
但是将自己的rknn模型参考官方的rknn_ssd_demo一直到Android上时，rknn_outputs_get函数出错
难道官方的例子里这个rknn_outputs_get只能获得coco数据集里的91类的输出，换了模型改了类别数后，rknn_outputs_get函数就不能被正常调用了？
697788=1917x4x91(91是coco数据集的类别数) 61344=1917x4x8（8是我们自己的类别数）
[attach]1212[/attach]

作者: jefferyzhang 时间: 2020-7-3 17:26
你这里改了，转换rknn模型时候也要改啊。。

作者: wang64 时间: 2020-7-6 19:51
我们改了呀，里面的类别数目我们改成自己的了，然后节点数目也改成自己的了，我们在pc端上用那个模型脚本测试过，可以检测到我们的物体，但是移植到android就不行了，api里面提供的那个rnn模型检测都可以的。

作者: wang64 时间: 2020-7-7 09:01

jefferyzhang 发表于 2020-7-3 17:26
你这里改了，转换rknn模型时候也要改啊。。

我们修改了里面的类别数与节点信息，在pc端上，那个rknn模型可以检测我们所需的物体，但是在android端上显示维度不匹配

作者: pjmhfhs 时间: 2020-7-7 10:02

jefferyzhang 发表于 2020-7-3 17:26
你这里改了，转换rknn模型时候也要改啊。。

具体是要改什么？哪些函数的参数？没看到有教程啊

作者: loi666 时间: 2020-7-7 16:46

jefferyzhang 发表于 2020-7-3 17:26
你这里改了，转换rknn模型时候也要改啊。。

我也按照训练改了,在pc端利用rknn可以进行目标检测,但是在android端将自己的 rknn模型替换调原rknn原模型后出现纬度不匹配的问题,怀疑是不是在生成jni文件时候输入固定成了定值

作者: wang64 时间: 2020-7-13 17:59

jefferyzhang 发表于 2020-7-3 17:26
你这里改了，转换rknn模型时候也要改啊。。

您好，我想问一下官方最新的rk3399pro开发板的NPU DRV版本是多少？

作者: jefferyzhang 时间: 2020-7-14 10:05

wang64 发表于 2020-7-13 17:59
您好，我想问一下官方最新的rk3399pro开发板的NPU DRV版本是多少？

1.3.2-1.3.3

作者: wang64 时间: 2020-7-14 11:07

jefferyzhang 发表于 2020-7-14 10:05
1.3.2-1.3.3

您好，我想再问一下使用1.3.0版本的rknn-toolkit转换得到的rknn模型和这个1.3.2-1.3.3的DRV匹配吗？

作者: jefferyzhang 时间: 2020-7-14 11:42

wang64 发表于 2020-7-14 11:07
您好，我想再问一下使用1.3.0版本的rknn-toolkit转换得到的rknn模型和这个1.3.2-1.3.3的DRV匹配吗？ ...

具体要看rknn toolkit文档，上头有详细的版本匹配说明。
理论上大版本是可匹配的 1.3.x 都可匹配。具体还是按文档上说的来

作者: wang64 时间: 2020-7-14 11:48

jefferyzhang 发表于 2020-7-14 11:42
具体要看rknn toolkit文档，上头有详细的版本匹配说明。
理论上大版本是可匹配的 1.3.x 都可匹配。具体还 ...

好的，谢谢

欢迎光临 Toybrick (https://t.rock-chips.com/)