Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
Uploaded yolov5 v6.0 wts file
  • Loading branch information
adujardin authored Jun 30, 2022
1 parent 4f04fb3 commit 8cdb122
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ This sample is a fork from [wang-xinyu/tensorrtx](https://github.com/wang-xinyu/

### 1. (Optional for first run) Generate .wts from pytorch with .pt

**This file has already been generated and can be downloaded [here](https://download.stereolabs.com/sample_custom_objects/yolov5s.wts.zip)** (and needs to be unzipped) to run the sample.
**This file has already been generated and can be downloaded [here](https://download.stereolabs.com/sample_custom_objects/yolov5s_v6.0.wts.zip)** (and needs to be unzipped) to run the sample.

This procedure can be applied to other models (such as `l` or `m` variants) or custom dataset trained model.

Expand All @@ -42,7 +42,7 @@ git clone -b v6.0 https://github.com/ultralytics/yolov5.git
wget https://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5s.pt
cp gen_wts.py {ultralytics}/yolov5
cd {ultralytics}/yolov5
python gen_wts.py yolov5s.pt
python gen_wts.py -w yolov5s.pt
# a file 'yolov5s.wts' will be generated.
```

Expand Down

0 comments on commit 8cdb122

Please sign in to comment.