Toybrick

自定义OP错误

zhanghui-aibee

新手上路

积分
29
发表于 2020-5-22 15:14:03    查看: 96|回复: 6 | [复制链接]    打印 | 显示全部楼层
Hi , 我们需要添加自定义OP,遇到一些问题,希望解答一下。
参考官方手册《Rockchip_Developer_Guide_RKNN_Toolkit_Custom_OP_CN》
遇到以下问题:
1.使用rknn-toolkit-V1.3.0工具,测试examples/common_function_demos/custom_op/rknn_custom_op_resize示例,
  运行python3 -m rknn.bin.custom_op --action build --op_path ./resize_area


  PC运行测试示例: 提示 E [str2op:135]Can not find op:RESIZEAREA

  3399pro板卡运行报错:
  

  测试代码:在转换模型时会把load_rknn注释掉。
  
2.在《Rockchip_User_Guide_RKNN_API_V1.3.0_CN》[img][/img]文档中,并没有加载自定义op的相关API? 如何在调用API的情况下,加载自定义OP?

谢谢!

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

超级版主

积分
4770
发表于 2020-5-26 08:29:37 | 显示全部楼层
1.3.2试过不?
回复

使用道具 举报

zhanghui-aibee

新手上路

积分
29
 楼主| 发表于 2020-5-26 10:58:42 | 显示全部楼层

没有,1.3.2对自定义算子有更新吗?
回复

使用道具 举报

zhanghui-aibee

新手上路

积分
29
 楼主| 发表于 2020-5-26 21:36:28 | 显示全部楼层

测试了一下,还是过不了,报相同的错误。
测试环境:v1.3.2
测试自定义op使用的是rknn-toolkit-1.3.2中examples/common_function_demos/custom_op/rknn_custom_op_resize/resize_area/ResizeArea.rknnop
测试源码:
  1. rknn = RKNN(verbose=False)
  2.     print("register_op:")
  3.     rknn.register_op('./resize_area/ResizeArea.rknnop')
  4.     print("end")

  5.     '''
  6.     rknn.load_tensorflow(tf_pb='./resize_area_test.pb',
  7.                          inputs=['input'],
  8.                          outputs=['resize_area_0'],
  9.                          input_size_list=[[32, 32, 3]])
  10.     rknn.build(do_quantization=False)
  11.     # rknn.export_rknn('./resize_area.rknn')
  12.     '''
  13.     print("load rknn mode:")
  14.     rknn.load_rknn('./resize_area.rknn')
  15.     print("load end")

  16.     print("init_runtime:")
  17.     rknn.init_runtime()
  18.     print("end")

  19.     img = cv2.imread('./dog_32x32.jpg')

  20.     outs = rknn.inference(inputs=[img])

  21.     out_img = outs[0].astype('uint8')
  22.     out_img = np.reshape(out_img, (64, 64, 3))
  23.     cv2.imwrite('./out.jpg', out_img)
复制代码


在rknn3399pro上报错与1.3.0相同:
  1. I NPUTransfer: Starting NPU Transfer Client, Transfer version 1.9.8 (40e4a8a@2020-01-02T09:16:20)
  2. E RKNNAPI: rknn_init,  recv(MsgLoadAck) fail, -9(ERROR_PIPE) != 368!
  3. E Catch exception when init runtime!
  4. E Traceback (most recent call last):
  5. E   File "rknn/api/rknn_base.py", line 1067, in rknn.api.rknn_base.RKNNBase.init_runtime
  6. E   File "rknn/api/rknn_runtime.py", line 326, in rknn.api.rknn_runtime.RKNNRuntime.build_graph
  7. E Exception: RKNN init failed. error code: RKNN_ERR_DEVICE_UNAVAILABLE
复制代码
回复

使用道具 举报

jefferyzhang

超级版主

积分
4770
发表于 2020-5-27 00:03:02 | 显示全部楼层
已经报给NPU Team了,等他们答复你
回复

使用道具 举报

zhanghui-aibee

新手上路

积分
29
 楼主| 发表于 2020-5-27 09:59:55 | 显示全部楼层
jefferyzhang 发表于 2020-5-27 00:03
已经报给NPU Team了,等他们答复你

好的,期待答复!感谢!
回复

使用道具 举报

zhanghui-aibee

新手上路

积分
29
 楼主| 发表于 5 天前 | 显示全部楼层
自己顶一下,不要沉下去
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表