From 25ec3e4a521386fc2e8ef039494ed45c56a642c4 Mon Sep 17 00:00:00 2001 From: felipemasuda Date: Thu, 17 Oct 2024 14:31:25 -0300 Subject: [PATCH 1/2] Update of instructions for using custom models in Yolov8 [README.md] --- yolov8/README.md | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) mode change 100644 => 100755 yolov8/README.md diff --git a/yolov8/README.md b/yolov8/README.md old mode 100644 new mode 100755 index 8bada0d4..912c3d53 --- a/yolov8/README.md +++ b/yolov8/README.md @@ -23,6 +23,12 @@ The tensorrt code is derived from [xiaocao-tian/yolov8_tensorrt](https://github. Currently, we support yolov8 - For yolov8 , download .pt from [https://github.com/ultralytics/assets/releases](https://github.com/ultralytics/assets/releases), then follow how-to-run in current page. + +## Custom models + +- Use ultralytics<=8.2.103 for training. +- Update kNumClass in [include/config.h](./include/config.h) to the number of classes in the custom model + ## Config - Choose the model n/s/m/l/x/n2/s2/m2/l2/x2/n6/s6/m6/l6/x6 from command line arguments. @@ -60,7 +66,6 @@ python gen_wts.py -w yolov5nu.pt -o yolov5nu.wts -t detect ### Detection ``` cd {tensorrtx}/yolov8/ -// update kNumClass in config.h if your model is trained on custom dataset mkdir build cd build cp {ultralytics}/ultralytics/yolov8.wts {tensorrtx}/yolov8/build From 0fac0bc3fac2036091e4311ad4f2f31237a21b4c Mon Sep 17 00:00:00 2001 From: felipemasuda <68308243+felipemasuda@users.noreply.github.com> Date: Fri, 18 Oct 2024 15:48:43 -0300 Subject: [PATCH 2/2] Update README.md Fix ultralytics<=8.2.103 --- yolov8/README.md | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/yolov8/README.md b/yolov8/README.md index 912c3d53..ef9a6a8c 100755 --- a/yolov8/README.md +++ b/yolov8/README.md @@ -17,6 +17,7 @@ The tensorrt code is derived from [xiaocao-tian/yolov8_tensorrt](https://github. - TensorRT 8.0+ - OpenCV 3.4.0+ +- ultralytics<=8.2.103 ## Different versions of yolov8 @@ -24,11 +25,6 @@ Currently, we support yolov8 - For yolov8 , download .pt from [https://github.com/ultralytics/assets/releases](https://github.com/ultralytics/assets/releases), then follow how-to-run in current page. -## Custom models - -- Use ultralytics<=8.2.103 for training. -- Update kNumClass in [include/config.h](./include/config.h) to the number of classes in the custom model - ## Config - Choose the model n/s/m/l/x/n2/s2/m2/l2/x2/n6/s6/m6/l6/x6 from command line arguments.