Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Update onnx>=1.12.0 (#10526)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-12-18 20:06:01 +01:00
1 parent 2c35c1b318
commit b2a0f1cdc5
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

export.py
+1 -1
View File
@@ -132,7 +132,7 @@ def export_torchscript(model, im, file, optimize, prefix=colorstr('TorchScript:'
@try_export @try_export
def export_onnx(model, im, file, opset, dynamic, simplify, prefix=colorstr('ONNX:')): def export_onnx(model, im, file, opset, dynamic, simplify, prefix=colorstr('ONNX:')):
# YOLOv5 ONNX export # YOLOv5 ONNX export
check_requirements('onnx') check_requirements('onnx>=1.12.0')
import onnx import onnx
LOGGER.info(f'\n{prefix} starting export with onnx {onnx.__version__}...') LOGGER.info(f'\n{prefix} starting export with onnx {onnx.__version__}...')
requirements.txt
+1 -1
View File
@@ -29,7 +29,7 @@ seaborn>=0.11.0
# Export ---------------------------------------------------------------------- # Export ----------------------------------------------------------------------
# coremltools>=6.0 # CoreML export # coremltools>=6.0 # CoreML export
# onnx>=1.9.0 # ONNX export # onnx>=1.12.0 # ONNX export
# onnx-simplifier>=0.4.1 # ONNX simplifier # onnx-simplifier>=0.4.1 # ONNX simplifier
# nvidia-pyindex # TensorRT export # nvidia-pyindex # TensorRT export
# nvidia-tensorrt # TensorRT export # nvidia-tensorrt # TensorRT export