Description
PPOCR模型onnx转换成rknn的过程中,报错 Unknow elem_type for softmax_2.tmp_0!,rknn-toolkit2 version: 2.3.2和rknn-toolkit2 version: 2.0.0版本都试了,都是一样的报错,请问怎么解决呢
python3 convert.py ../model/static_rec_model.onnx rk3568
I rknn-toolkit2 version: 2.3.2
--> Config model
done
--> Loading model
E load_onnx: Unknow elem_type for softmax_2.tmp_0!
I ===================== WARN(0) =====================
E rknn-toolkit2 version: 2.3.2
E load_onnx: Traceback (most recent call last):
File "rknn/api/rknn_log.py", line 344, in rknn.api.rknn_log.error_catch_decorator.error_catch_wrapper
File "rknn/api/rknn_base.py", line 1579, in rknn.api.rknn_base.RKNNBase.load_onnx
File "rknn/api/rknn_base.py", line 613, in rknn.api.rknn_base.RKNNBase._create_ir_and_inputs_meta
File "rknn/api/ir_graph.py", line 84, in rknn.api.ir_graph.IRGraph.init
File "rknn/api/ir_graph.py", line 669, in rknn.api.ir_graph.IRGraph.rebuild
File "rknn/api/rknn_log.py", line 95, in rknn.api.rknn_log.RKNNLog.e
ValueError: Unknow elem_type for softmax_2.tmp_0!
I ===================== WARN(0) =====================
E rknn-toolkit2 version: 2.3.2
Traceback (most recent call last):
File "rknn/api/rknn_log.py", line 344, in rknn.api.rknn_log.error_catch_decorator.error_catch_wrapper
File "rknn/api/rknn_base.py", line 1579, in rknn.api.rknn_base.RKNNBase.load_onnx
File "rknn/api/rknn_base.py", line 613, in rknn.api.rknn_base.RKNNBase._create_ir_and_inputs_meta
File "rknn/api/ir_graph.py", line 84, in rknn.api.ir_graph.IRGraph.init
File "rknn/api/ir_graph.py", line 669, in rknn.api.ir_graph.IRGraph.rebuild
File "rknn/api/rknn_log.py", line 95, in rknn.api.rknn_log.RKNNLog.e
ValueError: Unknow elem_type for softmax_2.tmp_0!
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "convert.py", line 68, in
ret = rknn.load_onnx(model=model_path)
File "/home/uto/work/ocr/anaconda3/envs/python3.8-tk2/lib/python3.8/site-packages/rknn/api/rknn.py", line 168, in load_onnx
return self.rknn_base.load_onnx(model, inputs, input_size_list, input_initial_val, outputs)
File "rknn/api/rknn_log.py", line 349, in rknn.api.rknn_log.error_catch_decorator.error_catch_wrapper
File "rknn/api/rknn_log.py", line 95, in rknn.api.rknn_log.RKNNLog.e
ValueError: Traceback (most recent call last):
File "rknn/api/rknn_log.py", line 344, in rknn.api.rknn_log.error_catch_decorator.error_catch_wrapper
File "rknn/api/rknn_base.py", line 1579, in rknn.api.rknn_base.RKNNBase.load_onnx
File "rknn/api/rknn_base.py", line 613, in rknn.api.rknn_base.RKNNBase._create_ir_and_inputs_meta
File "rknn/api/ir_graph.py", line 84, in rknn.api.ir_graph.IRGraph.init
File "rknn/api/ir_graph.py", line 669, in rknn.api.ir_graph.IRGraph.rebuild
File "rknn/api/rknn_log.py", line 95, in rknn.api.rknn_log.RKNNLog.e
ValueError: Unknow elem_type for softmax_2.tmp_0!