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