Skip to content

Commit

Permalink
yolov8_5u_det model download link
Browse files Browse the repository at this point in the history
  • Loading branch information
lindsayshuo committed Sep 19, 2024
1 parent 817ed26 commit 6794858
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions yolov8/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,12 @@ python gen_wts.py -w yolov8n.pt -o yolov8n.wts -t detect
// download https://github.com/lindsayshuo/yolov8_p2_tensorrtx/releases/download/VisDrone_train_yolov8x_p2_bs1_epochs_100_imgsz_1280_last/VisDrone_train_yolov8x_p2_bs1_epochs_100_imgsz_1280_last.pt (only for 10 cls p2 model)
python gen_wts.py -w VisDrone_train_yolov8x_p2_bs1_epochs_100_imgsz_1280_last.pt -o VisDrone_train_yolov8x_p2_bs1_epochs_100_imgsz_1280_last.wts -t detect (only for 10 cls p2 model)
// a file 'VisDrone_train_yolov8x_p2_bs1_epochs_100_imgsz_1280_last.wts' will be generated.
// For yolov8_5u_det model
// download https://github.com/ultralytics/assets/releases/yolov5nu.pt
python gen_wts.py -w yolov5nu.pt -o yolov5nu.wts -t detect
// a file 'yolov5nu.wts' will be generated.
```

2. build tensorrtx/yolov8 and run
Expand Down

0 comments on commit 6794858

Please sign in to comment.