xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 05:24:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix onnxslim version (#380)

Browse Source
This commit is contained in:
inisis 2024-07-30 15:16:55 +08:00 committed by GitHub
parent 0e51495ce6
commit cd2f79c702
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -5,7 +5,7 @@ onnxruntime==1.15.1
pycocotools==2.0.7 pycocotools==2.0.7
PyYAML==6.0.1 PyYAML==6.0.1
scipy==1.13.0 scipy==1.13.0
onnxslim==0.1.28 onnxslim==0.1.31
onnxruntime-gpu==1.18.0 onnxruntime-gpu==1.18.0
gradio==4.31.5 gradio==4.31.5
opencv-python==4.9.0.80 opencv-python==4.9.0.80

6
ultralytics/engine/exporter.py
View File

@ -355,9 +355,7 @@ class Exporter:
"""YOLOv8 ONNX export.""" """YOLOv8 ONNX export."""
requirements = ["onnx>=1.12.0"] requirements = ["onnx>=1.12.0"]
if self.args.simplify: if self.args.simplify:
requirements += ["onnxslim==0.1.28", "onnxruntime" + ("-gpu" if torch.cuda.is_available() else "")] requirements += ["onnxslim==0.1.31", "onnxruntime" + ("-gpu" if torch.cuda.is_available() else "")]
if ARM64:
check_requirements("cmake") # 'cmake' is needed to build onnxsim on aarch64
check_requirements(requirements) check_requirements(requirements)
import onnx # noqa import onnx # noqa
@ -744,7 +742,7 @@ class Exporter:
"onnx>=1.12.0", "onnx>=1.12.0",
"onnx2tf>=1.15.4,<=1.17.5", "onnx2tf>=1.15.4,<=1.17.5",
"sng4onnx>=1.0.1", "sng4onnx>=1.0.1",
"onnxslim==0.1.28", "onnxslim==0.1.31",
"onnx_graphsurgeon>=0.3.26", "onnx_graphsurgeon>=0.3.26",
"tflite_support", "tflite_support",
"flatbuffers>=23.5.26,<100", # update old 'flatbuffers' included inside tensorflow package "flatbuffers>=23.5.26,<100", # update old 'flatbuffers' included inside tensorflow package