提交 4e32b602 authored 作者: Glenn Jocher's avatar Glenn Jocher

update

上级 66d73e4c
# 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()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论