apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

minor fix

Browse Source
This commit is contained in:
AlexeyAB84 2022-07-25 04:27:28 +03:00
parent 82f26becad
commit 703c2ff21a
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/YOLOv7onnx.ipynb
View File

@ -359,7 +359,7 @@
"!python export.py --weights ./yolov7-tiny.pt \\\n", "!python export.py --weights ./yolov7-tiny.pt \\\n",
" --grid --end2end --simplify \\\n", " --grid --end2end --simplify \\\n",
" --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 \\\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" " # otherwise it is non-agnostic NMS"
], ],
"metadata": { "metadata": {

2
tools/YOLOv7trt.ipynb
View File

@ -421,7 +421,7 @@
"cell_type": "code", "cell_type": "code",
"source": [ "source": [
"# export temporary ONNX model for TensorRT converter\n", "# 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" "!ls"
], ],
"metadata": { "metadata": {