glenn-jocher commited on
Commit
4e32b60
·
1 Parent(s): 66d73e4
Files changed (1) hide show
  1. models/onnx_export.py +2 -2
models/onnx_export.py CHANGED
@@ -1,4 +1,4 @@
1
- # Exports a pytorch *.pt model to *.onnx format. Example usage:
2
  # $ export PYTHONPATH="$PWD"
3
  # $ python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
4
 
@@ -10,7 +10,7 @@ from models.common import *
10
 
11
  if __name__ == '__main__':
12
  parser = argparse.ArgumentParser()
13
- parser.add_argument('--weights', default='./weights/yolov5s.pt', help='model path RELATIVE to ./models/')
14
  parser.add_argument('--img-size', default=640, help='inference size (pixels)')
15
  parser.add_argument('--batch-size', default=1, help='batch size')
16
  opt = parser.parse_args()
 
1
+ # Exports a pytorch *.pt model to *.onnx format. Example usage (run from ./yolov5 directory):
2
  # $ export PYTHONPATH="$PWD"
3
  # $ python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
4
 
 
10
 
11
  if __name__ == '__main__':
12
  parser = argparse.ArgumentParser()
13
+ parser.add_argument('--weights', default='./weights/yolov5s.pt', help='weights path')
14
  parser.add_argument('--img-size', default=640, help='inference size (pixels)')
15
  parser.add_argument('--batch-size', default=1, help='batch size')
16
  opt = parser.parse_args()