Toybrick

标题: Pytorch->onnx->rknn 推理结果错误 [打印本页]

作者: heizhima 时间: 2020-6-5 17:34
标题: Pytorch->onnx->rknn 推理结果错误
本帖最后由 heizhima 于 2020-6-5 17:36 编辑

背景：yolo系列pytorch模型，依次转onnx再到rknn模型，但是rknn推理结果为None。

1. pytorch模型推理结果是正常的。
2. onnx模型可视化正常，且推理结果正常。
3. pytorch推理和rknn推理的预处理和后处理代码一致。
4. 转rknn模型，测试do_quantization为True/False，推理结果都是为None。
5. 确认channle_mean和channel_order，inference输入的NHWC和NCHW的方式。

通过打印rknn模型的输出和后处理数据，发现是有输出数据，推理结果为None是由于检测的框对应的分数太低（小于0.5）均被丢弃，这个阈值改成0.1结果依旧为None。猜想rknn模型的推理和pytorch模型的推理结果不一致，且相差很大，在阈值过滤中全被丢弃了。
由于没有报错信息，不知如何调试，请指教！

[attach]1170[/attach]

作者: jefferyzhang 时间: 2020-6-6 21:11
DRV版本不匹配，请先升级DRV到1.3.x，不然没什么好说的，能跑通你都是中大奖了

作者: heizhima 时间: 2020-6-8 11:14

jefferyzhang 发表于 2020-6-6 21:11
DRV版本不匹配，请先升级DRV到1.3.x，不然没什么好说的，能跑通你都是中大奖了 ...

感谢您的回复，还请问一下DRV版本需要到哪里升级配置，烦请您解答或发一下相关链接，谢谢。

作者: heizhima 时间: 2020-6-8 15:29

jefferyzhang 发表于 2020-6-6 21:11
DRV版本不匹配，请先升级DRV到1.3.x，不然没什么好说的，能跑通你都是中大奖了 ...

您好，我在论坛中找到相应的升级DRV的方法了。目前DRV的版本为 1.3.1，可是还是这样的问题。

作者: jefferyzhang 时间: 2020-6-8 15:36
逐层调试，看下trouble_shoot文档

作者: heizhima 时间: 2020-6-18 09:06

jefferyzhang 发表于 2020-6-8 15:36
逐层调试，看下trouble_shoot文档

您好，问题已解决，是onnx模型输入w和h的问题。这个issue可以关了。

作者: wokafu 时间: 2022-4-18 21:07

heizhima 发表于 2020-6-18 09:06
您好，问题已解决，是onnx模型输入w和h的问题。这个issue可以关了。

大佬，你好！我也碰到差不多类似的问题，请问您是怎么解决的呢？要注意哪个地方？

作者: heizhima 时间: 2022-4-21 16:25

wokafu 发表于 2022-4-18 21:07
大佬，你好！我也碰到差不多类似的问题，请问您是怎么解决的呢？要注意哪个地方？ ...

你好，你可以检查一下图片输入的大小设置哈

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