update
This commit is contained in:
1 parent
66d73e4cf3
commit
4e32b6029a
1 file changed
+2
-2
@@ -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"
|
# $ export PYTHONPATH="$PWD"
|
||||||
# $ python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
|
# $ python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
|
||||||
|
|
||||||
@@ -10,7 +10,7 @@ from models.common import *
|
|||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
parser = argparse.ArgumentParser()
|
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('--img-size', default=640, help='inference size (pixels)')
|
||||||
parser.add_argument('--batch-size', default=1, help='batch size')
|
parser.add_argument('--batch-size', default=1, help='batch size')
|
||||||
opt = parser.parse_args()
|
opt = parser.parse_args()
|
||||||
|
|||||||
Reference in new issue
Block a user