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"