RetinaFace onnx model conversion to rknn reports error AttributeError: module 'onnx' has no attribute 'load'

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.

Seems like your python dependencies are not setup correctly. Are you using Python v3.8 and the requirements here?