glenn-jocher
commited on
Commit
•
57c342a
1
Parent(s):
3e7c59a
Simplify ONNX inference command (#4405)
Browse files
export.py
CHANGED
@@ -76,7 +76,7 @@ def export_onnx(model, img, file, opset, train, dynamic, simplify):
|
|
76 |
except Exception as e:
|
77 |
print(f'{prefix} simplifier failure: {e}')
|
78 |
print(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')
|
79 |
-
print(f"{prefix} run --dynamic ONNX model inference with
|
80 |
except Exception as e:
|
81 |
print(f'{prefix} export failure: {e}')
|
82 |
|
|
|
76 |
except Exception as e:
|
77 |
print(f'{prefix} simplifier failure: {e}')
|
78 |
print(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')
|
79 |
+
print(f"{prefix} run --dynamic ONNX model inference with: 'python detect.py --weights {f}'")
|
80 |
except Exception as e:
|
81 |
print(f'{prefix} export failure: {e}')
|
82 |
|