Team2890/PhotonVision
4
0
Fork 0
mirror of https://github.com/PhotonVision/photonvision synced 2026-06-19 00:41:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add downgrade fix for ONNX error during RKNN conversion (#2136)

Browse Source
This commit is contained in:
Rikhil Chilka
2025-10-23 13:17:31 -04:00
committed by GitHub
parent 054ed8b6a1
commit 7cb3b7a37b
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/rknn-convert-tool/create_onnx.py
View File

@@ -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)

2
scripts/rknn-convert-tool/rknn_conversion.ipynb
View File

@@ -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"
]
},
{