mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update onnx_export.py usage

pull/143/head
Glenn Jocher 2020-06-19 10:15:08 -07:00
parent 80b82e8bde
commit e249792c2f
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
models/onnx_export.py
View File

@ -1,7 +1,6 @@
"""Exports a pytorch *.pt model to *.onnx format """Exports a pytorch *.pt model to *.onnx format
Usage: Usage:
import torch
$ export PYTHONPATH="$PWD" && python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1 $ export PYTHONPATH="$PWD" && python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
""" """