From 4e32b6029aff9d541c383cba18c9a789d2a3a277 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 5 Jun 2020 20:03:21 -0700 Subject: [PATCH] update --- models/onnx_export.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/onnx_export.py b/models/onnx_export.py index a78aab9..f1bd18c 100644 --- a/models/onnx_export.py +++ b/models/onnx_export.py @@ -1,4 +1,4 @@ -# Exports a pytorch *.pt model to *.onnx format. Example usage: +# Exports a pytorch *.pt model to *.onnx format. Example usage (run from ./yolov5 directory): # $ export PYTHONPATH="$PWD" # $ python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1 @@ -10,7 +10,7 @@ from models.common import * if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument('--weights', default='./weights/yolov5s.pt', help='model path RELATIVE to ./models/') + parser.add_argument('--weights', default='./weights/yolov5s.pt', help='weights path') parser.add_argument('--img-size', default=640, help='inference size (pixels)') parser.add_argument('--batch-size', default=1, help='batch size') opt = parser.parse_args()