From d489e9435be86aae84ec93c213a89e567b048610 Mon Sep 17 00:00:00 2001 From: Alexey Date: Mon, 25 Jul 2022 04:01:02 +0300 Subject: [PATCH] Update README.md --- README.md | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 35c30843..fa1bfdf0 100644 --- a/README.md +++ b/README.md @@ -156,19 +156,21 @@ python detect.py --weights yolov7.pt --conf 0.25 --img-size 640 --source inferen ## Export -**Pytorch to ONNX** Open In Colab +**Pytorch to ONNX with NMS (and inference)** Open In Colab ```shell python export.py --weights yolov7-tiny.pt --grid --end2end --simplify \ --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --max-wh 640 ``` -**Pytorch to TensorRT** Open In Colab +**Pytorch to TensorRT with NMS (and inference)** Open In Colab ```shell wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-tiny.pt python export.py --weights ./yolov7-tiny.pt --grid --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 git clone https://github.com/Linaom1214/tensorrt-python.git python ./tensorrt-python/export.py -o yolov7-tiny.onnx -e yolov7-tiny-nms.trt -p fp16 + +# example of inference in C++ https://github.com/Linaom1214/tensorrt-python/tree/main/yolov7/cpp ``` **Pytorch to TensorRT another way** Open In Colab
Expand