Fix logger error in DeepStream 6.0 / 6.0.1 + Change output classes format + Fixes
This commit is contained in:
@@ -46,30 +46,6 @@ Generate the ONNX model file (example for YOLOv8s)
|
||||
python3 export_yoloV8.py -w yolov8s.pt --dynamic
|
||||
```
|
||||
|
||||
**NOTE**: To simplify the ONNX model (DeepStream >= 6)
|
||||
|
||||
```
|
||||
--simplify
|
||||
```
|
||||
|
||||
**NOTE**: To use dynamic batch-size (DeepStream >= 6)
|
||||
|
||||
```
|
||||
--dynamic
|
||||
```
|
||||
|
||||
**NOTE**: To use implicit batch-size (example for batch-size = 4)
|
||||
|
||||
```
|
||||
--batch 4
|
||||
```
|
||||
|
||||
**NOTE**: If you are using DeepStream 5.1, remove the `--dynamic` arg and use opset 12 or lower. The default opset is 16.
|
||||
|
||||
```
|
||||
--opset 12
|
||||
```
|
||||
|
||||
**NOTE**: To change the inference size (defaut: 640)
|
||||
|
||||
```
|
||||
@@ -91,6 +67,30 @@ or
|
||||
-s 1280 1280
|
||||
```
|
||||
|
||||
**NOTE**: To simplify the ONNX model (DeepStream >= 6.0)
|
||||
|
||||
```
|
||||
--simplify
|
||||
```
|
||||
|
||||
**NOTE**: To use dynamic batch-size (DeepStream >= 6.1)
|
||||
|
||||
```
|
||||
--dynamic
|
||||
```
|
||||
|
||||
**NOTE**: To use implicit batch-size (example for batch-size = 4)
|
||||
|
||||
```
|
||||
--batch 4
|
||||
```
|
||||
|
||||
**NOTE**: If you are using DeepStream 5.1, remove the `--dynamic` arg and use opset 12 or lower. The default opset is 16.
|
||||
|
||||
```
|
||||
--opset 12
|
||||
```
|
||||
|
||||
#### 5. Copy generated files
|
||||
|
||||
Copy the generated ONNX model file and labels.txt file (if generated) to the `DeepStream-Yolo` folder.
|
||||
|
||||
Reference in New Issue
Block a user