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()