Browse Source

update onnx_export.py usage

5.0
Glenn Jocher 4 years ago
parent
commit
e249792c2f
1 changed files with 0 additions and 1 deletions
  1. +0
    -1
      models/onnx_export.py

+ 0
- 1
models/onnx_export.py View File

@@ -1,7 +1,6 @@
"""Exports a pytorch *.pt model to *.onnx format

Usage:
import torch
$ export PYTHONPATH="$PWD" && python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
"""


Loading…
Cancel
Save