Faster rcnn 无法得到正确的预测结果

[复制链接] · 发表于 2020-6-30 22:15:33

你好，我用rknn tools跑我训练的Faster rcnn caffe格式的模型在RK1080 NPU上，该模型经过8位定点量化后，无法得到正确的预测结果，请问这是什么原因呢？
我发现rknn的proposal层无im_info输入，请问默认的im_info是什么尺寸呢？

只看该作者 · 发表于 2020-7-1 08:57:04

不量化是否能得到正确结果？

只看该作者 · 发表于 2020-7-1 11:49:26

你好，我是在ubuntu x86-64平台上模拟rk1808 npu进行测试，选择不量化操作后，会出现如下错误：

W [set_chip_platform_env:187]evaluate model on RK1808

Check model fail

E Catch exception when init runtime!

E Traceback (most recent call last):

E   File "rknn/api/rknn_base.py", line 1067, in rknn.api.rknn_base.RKNNBase.init_runtime

E   File "rknn/api/rknn_runtime.py", line 326, in rknn.api.rknn_runtime.RKNNRuntime.build_graph

E Exception: RKNN init failed. error code: RKNN_ERR_MODEL_INVALID

Init runtime environment failed
复制代码

所以无法进行非量化实验，不知道是什么原因？

只看该作者 · 发表于 2022-1-5 10:36:42

1.RKNN的faster-rcnn转换的时候把im_info去除了
2.im_info是尺寸的比例，比如你的输入图像是512，而输入模型的尺寸是600，那im_info就是600/512
3.RKNN可以量化faster-rcnn，但是在roi polling层的时候会有较大的误差，同时不量化的时候，也有问题，可能是RKNN官方对faster-rcnn的支持不太友好