rknn转出后推理错误

[复制链接] · 发表于 2022-4-19 10:20:10

1. onnx推理结果跟pytorch推理结果一致，onnx转rknn后，rknn推理完成错误，打开rknn的各个卷积的weights看起来又是对的，在运行调试的时候提示：W Inputs should be placed in a list, like [img1, img2], both the img1 and img2 are ndarray

麻烦各位老师指点迷津，谢谢了！

只看该作者 · 发表于 2022-4-19 10:34:54

warning不用在意

只看该作者 · 发表于 2022-4-19 10:49:08

onnx推理是正确的，rknn推理完全不正确，请问这个一般是哪个步骤错误导致的呢？大佬！
打开rknn文件，各个层的weight参数感觉又是对的

只看该作者 · 发表于 2022-4-19 13:23:00

结题了，输入的时候图片处理得不对，各种变换处理后，推理正确了！在inference的参数指定data_format="nchw"

rknn转出后推理错误

本帖子中包含更多资源