thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #290 from faizan1234567/yolov7_docs_updated

Browse Source 
yolov7 docs updated
This commit is contained in:
Marcos Luciano
2023-01-26 07:50:43 -03:00
committed by GitHub
parent 71541cbb80 51615b6709
commit f1cd701247
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docs/YOLOv7.md
View File

@@ -36,7 +36,11 @@ wget hhttps://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt
**NOTE**: You can use your custom model, but it is important to keep the YOLO model reference (`yolov7_`) in you `cfg` and `weights`/`wts` filenames to generate the engine correctly. **NOTE**: You can use your custom model, but it is important to keep the YOLO model reference (`yolov7_`) in you `cfg` and `weights`/`wts` filenames to generate the engine correctly.
#### 4. Convert model #### 4. Reparameterize your model
[YOLOv7](https://github.com/WongKinYiu/yolov7/releases/) and it's variants can't be directly converted to engine file. Therefore, you will have to reparameterize your model using the code [here](https://github.com/WongKinYiu/yolov7/blob/main/tools/reparameterization.ipynb). Make sure to convert your checkpoints in yolov7 repository, and then save your reparmeterized checkpoints for conversion in the next step.
#### 5. Convert model
Generate the `cfg` and `wts` files (example for YOLOv7) Generate the `cfg` and `wts` files (example for YOLOv7)
@@ -65,7 +69,7 @@ or
-s 1280 1280 -s 1280 1280
``` ```
#### 5. Copy generated files #### 6. Copy generated files
Copy the generated `cfg` and `wts` files to the `DeepStream-Yolo` folder. Copy the generated `cfg` and `wts` files to the `DeepStream-Yolo` folder.