minor fix

kweston · Jul 25, 2022 · 703c2ff · 703c2ff
1 parent 82f26be
commit 703c2ff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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": {