Fix logger error in DeepStream 6.0 / 6.0.1 + Change output classes format + Fixes
This commit is contained in:
@@ -47,37 +47,13 @@ Generate the ONNX model file (example for YOLOv6-S 4.0)
|
||||
python3 export_yoloV6.py -w yolov6s.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 13.
|
||||
|
||||
```
|
||||
--opset 12
|
||||
```
|
||||
|
||||
**NOTE**: To convert a P6 model
|
||||
|
||||
```
|
||||
--p6
|
||||
```
|
||||
|
||||
**NOTE**: To change the inference size (defaut: 640)
|
||||
**NOTE**: To change the inference size (defaut: 640 / 1280 for `--p6` models)
|
||||
|
||||
```
|
||||
-s SIZE
|
||||
@@ -98,6 +74,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 13.
|
||||
|
||||
```
|
||||
--opset 12
|
||||
```
|
||||
|
||||
#### 5. Copy generated file
|
||||
|
||||
Copy the generated ONNX model file to the `DeepStream-Yolo` folder.
|
||||
|
||||
Reference in New Issue
Block a user