From 7cb3b7a37bd0da33c9a90be40924c2a3b3c391d0 Mon Sep 17 00:00:00 2001 From: Rikhil Chilka <63861512+boomermath@users.noreply.github.com> Date: Thu, 23 Oct 2025 13:17:31 -0400 Subject: [PATCH] Add downgrade fix for ONNX error during RKNN conversion (#2136) --- scripts/rknn-convert-tool/create_onnx.py | 7 +++++-- scripts/rknn-convert-tool/rknn_conversion.ipynb | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/scripts/rknn-convert-tool/create_onnx.py b/scripts/rknn-convert-tool/create_onnx.py index c568e2d57..c9b3b1133 100644 --- a/scripts/rknn-convert-tool/create_onnx.py +++ b/scripts/rknn-convert-tool/create_onnx.py @@ -82,7 +82,8 @@ def run_onnx_conversion_yolov5(model_path): "-r", os.path.join(ultralytics_folder_name_yolov5, "requirements.txt"), "torch<2.6.0", - "onnx", + "onnx==1.18.0", + "onnxscript", ] ) @@ -121,7 +122,9 @@ def run_onnx_conversion_yolov5(model_path): def run_onnx_conversion_no_anchor(model_path): check_or_clone_rockchip_repo(yolo_non_anchor_repo) - run_pip_install_or_else_exit(["-e", ultralytics_default_folder_name, "onnx"]) + run_pip_install_or_else_exit( + ["-e", ultralytics_default_folder_name, "onnx==1.18.0", "onnxscript"] + ) sys.path.insert(0, os.path.abspath(ultralytics_default_folder_name)) model_abs_path = os.path.abspath(model_path) diff --git a/scripts/rknn-convert-tool/rknn_conversion.ipynb b/scripts/rknn-convert-tool/rknn_conversion.ipynb index e330677df..eb07261ae 100644 --- a/scripts/rknn-convert-tool/rknn_conversion.ipynb +++ b/scripts/rknn-convert-tool/rknn_conversion.ipynb @@ -109,7 +109,7 @@ "\n", "#### Automatic installation\n", "\n", - "Please run `pip` below. If it does not work, refer to the instructions for manual installation.\n" + "Please run `pip` below. If it does not work, refer to the instructions for manual installation. You may need to restart your session after running the command below.\n" ] }, {