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))

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

  5. print('--> Init runtime environment of arcface')
  6. ret = net.init_runtime()
  7. if ret != 0:
  8.     print('Init runtime environment failed')
  9.     exit(ret)
  10. 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