From 000b8482dd59cdf3cc2cf9577445775317442796 Mon Sep 17 00:00:00 2001
From: Zheng-Bicheng <58363586+Zheng-Bicheng@users.noreply.github.com>
Date: Wed, 8 Feb 2023 11:13:44 +0800
Subject: [PATCH] [Bug Fix] Update ppyolov8 rknpu2 export config (#1256)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

更新ppyolov8 模型转换config
---
 tools/rknpu2/config/yolov8_n_quantized.yaml   |  8 ++++++++
 tools/rknpu2/config/yolov8_n_unquantized.yaml | 12 ++++++++++--
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/tools/rknpu2/config/yolov8_n_quantized.yaml b/tools/rknpu2/config/yolov8_n_quantized.yaml
index c0811864f0..bac5b8c20d 100644
--- a/tools/rknpu2/config/yolov8_n_quantized.yaml
+++ b/tools/rknpu2/config/yolov8_n_quantized.yaml
@@ -1,5 +1,13 @@
 mean:
+  -
+    - 0
+    - 0
+    - 0
 std:
+  -
+    - 255
+    - 255
+    - 255
 model_path: ./yolov8_n_500e_coco/yolov8_n_500e_coco.onnx
 outputs_nodes:
   - 'p2o.Mul.119'
diff --git a/tools/rknpu2/config/yolov8_n_unquantized.yaml b/tools/rknpu2/config/yolov8_n_unquantized.yaml
index c1a777e1d7..3d1a0f0534 100644
--- a/tools/rknpu2/config/yolov8_n_unquantized.yaml
+++ b/tools/rknpu2/config/yolov8_n_unquantized.yaml
@@ -1,9 +1,17 @@
 mean:
+  -
+    - 0
+    - 0
+    - 0
 std:
+  -
+    - 255
+    - 255
+    - 255
 model_path: ./yolov8_n_500e_coco/yolov8_n_500e_coco.onnx
 outputs_nodes:
-  - 'p2o.Div.1'
+  - 'p2o.Mul.1'
   - 'p2o.Concat.49'
 do_quantization: False
-dataset: "./dataset.txt"
+dataset: "./yolov8_n_500e_coco/dataset.txt"
 output_folder: "./yolov8_n_500e_coco"