mirror of https://github.com/WongKinYiu/yolov7.git
Update README.md (#850)
parent
8035ee6db9
commit
8b616af63a
deploy/triton-inference-server
|
@ -11,6 +11,9 @@ There are no additional dependencies needed to run this deployment, except a wor
|
||||||
See https://github.com/WongKinYiu/yolov7#export for more info.
|
See https://github.com/WongKinYiu/yolov7#export for more info.
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
|
#install onnx-simplifier not listed in general yolov7 requirements.txt
|
||||||
|
pip3 install onnx-simplifier
|
||||||
|
|
||||||
# Pytorch Yolov7 -> ONNX with grid, EfficientNMS plugin and dynamic batch size
|
# Pytorch Yolov7 -> ONNX with grid, EfficientNMS plugin and dynamic batch size
|
||||||
python export.py --weights ./yolov7.pt --grid --end2end --dynamic-batch --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640
|
python export.py --weights ./yolov7.pt --grid --end2end --dynamic-batch --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640
|
||||||
# ONNX -> TensorRT with trtexec and docker
|
# ONNX -> TensorRT with trtexec and docker
|
||||||
|
|
Loading…
Reference in New Issue