Toybrick

标题: 模型转换为rknn后inference结果为0 [打印本页]
作者: yaowei    时间: 2019-5-8 10:32
标题: 模型转换为rknn后inference结果为0
可以从哪些点、哪些途径查找错误在哪里呢
作者: troy    时间: 2019-5-9 08:40
问题描述需要精准一点，可以提供一下你的环境，RKNN版本，转换什么模型，如何转换的，如何推理的，这样别人才好回答你的问题。
作者: yaowei    时间: 2019-5-9 11:43
本帖最后由 yaowei 于 2019-5-9 11:45 编辑

RKNN版本：0.9.9

转换的模型：pytorch->onnx->rknn

onnx->rknn代码：
  1. from rknn.api import RKNN
  2. rknn = RKNN()
  3. rknn.config(channel_mean_value='127.5 127.5 127.5 128', reorder_channel='2 1 0')
  4. print('--> Loading model')
  5. rknn.load_onnx(model='./resnet.onnx')
  6. print('done')
  7. print('--> Building model')
  8. rknn.build(do_quantization=False)
  9. print('done')
  10. rknn.export_rknn('./resnet.rknn')
  11. rknn.release()
复制代码
作者: yaowei    时间: 2019-5-9 11:48
本帖最后由 yaowei 于 2019-5-15 15:43 编辑

rknn推理：

  1. img = cv2.imread('test.png')
  2. img = cv2.resize(img, (128,128))

  4. net = RKNN()
  5. net.load_rknn('./resnet.rknn')

  7. print('--> Init runtime environment of arcface')
  8. ret = net.init_runtime()
  9. if ret != 0:
  10.     print('Init runtime environment failed')
  11.     exit(ret)
  12. print('done')
复制代码output = net.inference(inputs=[img])

作者: yaowei    时间: 2019-5-9 11:52
推理的输出output都是0，我在ubuntu下未转换的原版权重测试过，是没问题的；除了pytorch->onnx的问题以外，哪里可能有问题呢
作者: xiaqing10    时间: 2019-5-15 15:18
net.load_rknn('.resnet.rknn') ?
这行确定没有问题?
作者: yaowei    时间: 2019-5-15 15:42
哦，对不起，发帖的时候编辑过，多删了东西，肯定是可以运行的，我再编辑一下



欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3