rockchip-linux / rknn-toolkit

BSD 3-Clause "New" or "Revised" License
771 stars 175 forks source link

Problem with ONNX in model resnet50v2 #428

Open bielolopez opened 5 months ago

bielolopez commented 5 months ago

Hello. I can't find the problem why it doesn't raise the model, I don't know if it is compatibility, but from what I see and it is torch although torchvision version 0.4.0 is not found for aarch4. All suggestions will be well received. Thank you.

log_feedback_to_the_rknn_toolkit_dev_team.log

sudo pip3 list Package Version


absl-py 2.0.0 astor 0.8.1 bleach 3.1.2 cachetools 4.2.4 certifi 2018.8.24 chardet 3.0.4 click 8.1.7 decorator 4.3.0 defusedxml 0.5.0 devscripts 2.19.5+deb10u1 dill 0.2.8.2 entrypoints 0.3 Flask 1.0.2 flatbuffers 1.10 future 0.18.3 gast 0.2.2 google-auth 1.35.0 google-auth-oauthlib 0.4.6 google-pasta 0.2.0 grpcio 1.60.0 h5py 2.8.0 html5lib 1.0.1 idna 2.6 importlib-metadata 6.7.0 iotop 0.6 ipykernel 4.9.0 ipython 5.8.0 ipython_genutils 0.2.0 itsdangerous 2.1.2 Jinja2 3.1.3 joblib 1.3.2 jsonschema 2.6.0 jupyter-client 5.2.3 jupyter-core 4.4.0 Keras-Applications 1.0.8 Keras-Preprocessing 1.1.2 lmdb 0.93 lxml 4.3.2 Markdown 3.4.4 MarkupSafe 2.1.3 mistune 0.8.4 nbconvert 5.4.0 nbformat 4.4.0 networkx 1.11 notebook 5.7.8 numpy 1.16.1 oauthlib 3.2.2 onnx 1.6.0 onnx-tf 1.2.1 opencv-python 4.4.0.46 opt-einsum 3.3.0 pandocfilters 1.4.2 pexpect 4.6.0 pickleshare 0.7.5 Pillow 5.3.0 pip 23.3.2 ply 3.11 prometheus-client 0.6.0 prompt-toolkit 1.0.15 protobuf 3.11.2 psutil 5.6.2 pyasn1 0.5.1 pyasn1-modules 0.3.0 pycairo 1.16.2 Pygments 2.3.1 PyGObject 3.30.4 python-dateutil 2.7.3 PyYAML 6.0.1 pyzmq 17.1.2 requests 2.22.0 requests-oauthlib 1.3.1 rknn-toolkit 1.7.1 rsa 4.9 ruamel.yaml 0.15.81 scikit-learn 1.0.2 scipy 1.1.0 Send2Trash 1.5.0 setuptools 68.0.0 simplegeneric 0.8.1 six 1.12.0 sklearn 0.0 ssh-import-id 5.7 tensorboard 2.0.2 tensorflow 2.0.0 tensorflow-estimator 2.0.1 termcolor 2.3.0 terminado 0.8.1 testpath 0.4.2 threadpoolctl 3.1.0 torch 1.5.1 torchvision 0.6.1 tornado 5.1.1 traitlets 4.3.2 typing_extensions 4.7.1 urllib3 1.24.1 wcwidth 0.1.7 webencodings 0.5.1 Werkzeug 2.2.3 wheel 0.42.0 wrapt 1.16.0 zipp 3.15.0

eRaul commented 5 months ago

Judging from the error log, the torch version may be too low. You can try updating the torch version to 1.10.0 or 1.13.0. It is recommended that convert model on an x86 Linux PC. On the arm platform, the versions supported by many dependent modules are relatively limited.