Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

update onnx_export.py usage

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-06-19 10:15:08 -07:00
1 parent 80b82e8bde
commit e249792c2f
1 file changed
-1
Download Patch File Download Diff File Expand all files Collapse all files
models/onnx_export.py
-1
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
""" """