From d10b1835c7f2e9fd5555298e9a4e29ae41601ffe Mon Sep 17 00:00:00 2001 From: Alexey Date: Mon, 25 Jul 2022 05:12:40 +0300 Subject: [PATCH] Update README.md --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index fa1bfdf0..4ad5fe3d 100644 --- a/README.md +++ b/README.md @@ -15,7 +15,7 @@ Implementation of paper - [YOLOv7: Trainable bag-of-freebies sets new state-of-t ## Web Demo -- Integrated into [Huggingface Spaces 🤗](https://huggingface.co/spaces/akhaliq/yolov7) using [Gradio](https://github.com/gradio-app/gradio). Try out the Web Demo [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/akhaliq/yolov7) +- Integrated into [Huggingface Spaces 🤗](https://huggingface.co/spaces/akhaliq/yolov7) using Gradio. Try out the Web Demo [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/akhaliq/yolov7) ## Performance @@ -159,14 +159,14 @@ python detect.py --weights yolov7.pt --conf 0.25 --img-size 640 --source inferen **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 + --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 --max-wh 640 ``` **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 +python export.py --weights ./yolov7-tiny.pt --grid --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 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