haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update of instructions for using custom models in Yolov8 [README.md] (#1590)

Browse Source 
* Update of instructions for using custom models in Yolov8 [README.md]

* Update README.md

Fix  ultralytics<=8.2.103
This commit is contained in:
felipemasuda 2024-10-21 01:48:45 -03:00 committed by GitHub
parent 4ce6f20693
commit 0f75fd75f8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
yolov8/README.md Normal file → Executable file
View File

@ -17,12 +17,14 @@ 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
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.
## 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 +62,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