Fix logger error in DeepStream 6.0 / 6.0.1 + Change output classes format + Fixes

docs/YOLONAS.md

@@ -46,30 +46,6 @@ Generate the ONNX model file (example for YOLO-NAS S)
 python3 export_yolonas.py -m yolo_nas_s -w yolo_nas_s_coco.pth --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 14.
-
-```
---opset 12
-```
-
 **NOTE**: Model names
 
 ```
@@ -88,6 +64,18 @@ or
 -m yolo_nas_l
 ```
 
+**NOTE**: Number of classes (example for 80 classes)
+
+```
+-n 80
+```
+
+or
+
+```
+--classes 80
+```
+
 **NOTE**: To change the inference size (defaut: 640)
 
 ```
@@ -109,6 +97,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 14.
+
+```
+--opset 12
+```
+
 #### 5. Copy generated file
 
 Copy the generated ONNX model file to the `DeepStream-Yolo` folder.