Update onnx>=1.12.0 (#10526)
This commit is contained in:
1 parent
2c35c1b318
commit
b2a0f1cdc5
2 files changed
+2
-2
No files matched your search
@@ -132,7 +132,7 @@ def export_torchscript(model, im, file, optimize, prefix=colorstr('TorchScript:'
|
||||
@try_export
|
||||
def export_onnx(model, im, file, opset, dynamic, simplify, prefix=colorstr('ONNX:')):
|
||||
# YOLOv5 ONNX export
|
||||
check_requirements('onnx')
|
||||
check_requirements('onnx>=1.12.0')
|
||||
import onnx
|
||||
|
||||
LOGGER.info(f'\n{prefix} starting export with onnx {onnx.__version__}...')
|
||||
|
||||
+1
-1
@@ -29,7 +29,7 @@ seaborn>=0.11.0
|
||||
|
||||
# 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
|
||||
# nvidia-pyindex # TensorRT export
|
||||
# nvidia-tensorrt # TensorRT export
|
||||
|
||||
Reference in new issue
Block a user