Following model zoo example guide to setup RetinaFace demo,
When onnx model conversion to rknn reports error.
$ python convert.py …/model/RetinaFace_mobile320.onnx rk3566
I rknn-toolkit2 version: 2.3.2
–> Config model
done
–> Loading model
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 1532, in rknn.api.rknn_base.RKNNBase.load_onnx
AttributeError: module ‘onnx’ has no attribute ‘load’
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 1532, in rknn.api.rknn_base.RKNNBase.load_onnx
AttributeError: module ‘onnx’ has no attribute ‘load’
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File “convert.py”, line 50, in
ret = rknn.load_onnx(model=model_path)
File “/home/flink/anaconda3/envs/rknn/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 1532, in rknn.api.rknn_base.RKNNBase.load_onnx
AttributeError: module ‘onnx’ has no attribute ‘load’
Did anyone facing same problem too?
Looking for help or advise, thanks.