Commit
•
e249792
1
Parent(s):
80b82e8
update onnx_export.py usage
Browse files- models/onnx_export.py +0 -1
models/onnx_export.py
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
"""Exports a pytorch *.pt model to *.onnx format
|
| 2 |
|
| 3 |
Usage:
|
| 4 |
-
import torch
|
| 5 |
$ export PYTHONPATH="$PWD" && python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
|
| 6 |
"""
|
| 7 |
|
|
|
|
| 1 |
"""Exports a pytorch *.pt model to *.onnx format
|
| 2 |
|
| 3 |
Usage:
|
|
|
|
| 4 |
$ export PYTHONPATH="$PWD" && python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
|
| 5 |
"""
|
| 6 |
|