From 703c2ff21a9c7acbacf912b99754bccb94241af5 Mon Sep 17 00:00:00 2001
From: AlexeyAB84 <alexeyab84@gmail.com>
Date: Mon, 25 Jul 2022 04:27:28 +0300
Subject: [PATCH] minor fix

---
 tools/YOLOv7onnx.ipynb | 2 +-
 tools/YOLOv7trt.ipynb  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/YOLOv7onnx.ipynb b/tools/YOLOv7onnx.ipynb
index 1f6fc32..5fd501e 100644
--- a/tools/YOLOv7onnx.ipynb
+++ b/tools/YOLOv7onnx.ipynb
@@ -359,7 +359,7 @@
         "!python export.py --weights ./yolov7-tiny.pt \\\n",
         "        --grid --end2end --simplify \\\n",
         "        --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 \\\n",
-        "        --max-wh 640 # For onnxruntime, you need to specify this value as an integer, when it is 0 it means agnostic NMS, \n",
+        "        --img-size 640 640 --max-wh 640 # For onnxruntime, you need to specify this value as an integer, when it is 0 it means agnostic NMS, \n",
         "                     # otherwise it is non-agnostic NMS"
       ],
       "metadata": {
diff --git a/tools/YOLOv7trt.ipynb b/tools/YOLOv7trt.ipynb
index 145fbba..63e0865 100644
--- a/tools/YOLOv7trt.ipynb
+++ b/tools/YOLOv7trt.ipynb
@@ -421,7 +421,7 @@
       "cell_type": "code",
       "source": [
         "# export temporary ONNX model for TensorRT converter\n",
-        "!python export.py --weights ./yolov7-tiny.pt --grid --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35\n",
+        "!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\n",
         "!ls"
       ],
       "metadata": {