Fix YOLOv8 accuracy

This commit is contained in:
Marcos Luciano
2023-02-09 00:41:35 -03:00
parent 940d244ad4
commit 2d6b92eb64
9 changed files with 43 additions and 10 deletions

View File

@@ -82,7 +82,7 @@ Open the `DeepStream-Yolo` folder and compile the lib
### Edit the config_infer_primary_ppyoloe_plus file
Edit the `config_infer_primary_ppyoloe_plus.txt` file according to your model (example for PP-YOLOE+_s)
Edit the `config_infer_primary_ppyoloe_plus.txt` file according to your model (example for PP-YOLOE+_s with 80 classes)
```
[property]
@@ -90,6 +90,8 @@ Edit the `config_infer_primary_ppyoloe_plus.txt` file according to your model (e
custom-network-config=ppyoloe_plus_crn_s_80e_coco.cfg
model-file=ppyoloe_plus_crn_s_80e_coco.wts
...
num-detected-classes=80
...
```
**NOTE**: If you use the **legacy** model, you should edit the `config_infer_primary_ppyoloe.txt` file.
@@ -129,3 +131,5 @@ config-file=config_infer_primary_ppyoloe_plus.txt
```
deepstream-app -c deepstream_app_config.txt
```
**NOTE**: For more information about custom models configuration (`batch-size`, `network-mode`, etc), please check the [`docs/customModels.md`](customModels.md) file.