From 3c995974f50f52a6f372835113e2127d96737fa3 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 25 Jul 2021 13:45:57 +0200 Subject: [PATCH] Add `export.py` ONNX inference suggestion --- export.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/export.py b/export.py index ceee15e1644f..c98e92d972c6 100644 --- a/export.py +++ b/export.py @@ -76,6 +76,7 @@ def export_onnx(model, img, file, opset, train, dynamic, simplify): except Exception as e: print(f'{prefix} simplifier failure: {e}') print(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)') + print(f"{prefix} run --dynamic ONNX model inference with detect.py: 'python detect.py --weights {f}'") except Exception as e: print(f'{prefix} export failure: {e}') @@ -94,7 +95,7 @@ def export_coreml(model, img, file): model.save(f) print(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)') except Exception as e: - print(f'{prefix} export failure: {e}') + print(f'\n{prefix} export failure: {e}') def run(weights='./yolov5s.pt', # weights path