|
本帖最后由 wjzhoumm 于 2021-5-19 19:20 编辑
RKNN版本:1.6.0
TensorFlow版本:1.14.0
运行官方给的自定义算子的demo,可以正确生成对应的算子文件也就是rknnop,但是在试验其他的模型时会出错,比如试验tf.nn.selu算子时,可以正常生成Selu.rknnop算子文件,但是在PC模拟器上运行主程序却会报错,主程序代码如下:- from rknn.api import RKNN
- import numpy as np
- import cv2
- if __name__ == '__main__':
- rknn = RKNN(verbose=True)
- rknn.register_op('./selu/Selu.rknnop')
- rknn.load_tensorflow('selu.pb',
- inputs=['x'],
- outputs=['output'],
- input_size_list=[[32, 32, 3]])
- rknn.build(do_quantization=False)
- rknn.init_runtime()
- rknn.release()
报错内容如下:
- I [register_client_ops:5359]register_client_ops num=1
- I [register_client_ops:5372]op name=Selu�r reg_name=SELU�R reg_id=65542
- I [load_custom_op_lib:485]vx_kernel_Selu�r_list=0
- I [register_client_ops:5377]register_client_ops finish
- I [setup_nodes:4413]name=output_2
- E [str2op:130]Can not find op:SELU
- I [match_custom_op:141]try match SELU�R 65542
- E [match_custom_op:148]Can not find custom op:SELU
- E [setup_nodes:4434]match custom op fail
- I [remove_client_ops:5388]remove_client_ops num=1
- I [remove_client_ops:5394]remove op Selu�r reg_name=SELU�R reg_id=65542
- I [remove_client_ops:5399]remove_client_ops finish
- Create RKNN model fail, error=0
- Segmentation fault
算子的名字不知道为什么会变成了乱码,可能是这里导致的错误。
然后这个主程序在开发板rk3399pro上面可以正常运行,并且可以正确生成rknn模型,难道说这个主程序只能运行在开发板上吗?
请问这是什么导致的?
|
|