rknn-toolkit2
rknn-toolkit2 copied to clipboard
rknn toolkit2中进行onnx模型转换rknn,手动修改onnx输入输出,该模型使用onnx加载推理没有问题,但是转为rknn模型load报错,输入本来是1,3,32,192,报错输入0,3,0,0
rknn toolkit2中进行onnx模型转换rknn,手动修改onnx输入输出,该模型使用onnx加载推理没有问题,但是转为rknn模型load报错,输入本来是1,3,32,192,报错输入0,3,0,0 请问rknn现在可以支持lstm了吗?
E load_onnx: Traceback (most recent call last): E load_onnx: File "rknn/api/rknn_base.py", line 1182, in rknn.api.rknn_base.RKNNBase.load_onnx E load_onnx: File "rknn/api/rknn_base.py", line 646, in rknn.api.rknn_base.RKNNBase._create_ir_and_inputs_meta E load_onnx: File "rknn/api/rknn_log.py", line 113, in rknn.api.rknn_log.RKNNLog.e E load_onnx: ValueError: Invalid input_shape = [0, 3, 0, 0] for input 0!