实现onnx模型转rknn模型

008PaddleOCR_ONNX转rknn.py

@@ -9,14 +9,14 @@ rknn = RKNN()
 # 配置参数（关键！）
 rknn.config(
     target_platform="rk3588",          # 根据实际芯片型号修改
-    mean_values=[[0.485, 0.456, 0.406]],       # YOLOv8 输入为 0-255，无需归一化
-    std_values=[[255, 255, 255]],  # 输入数据除以 255（即 0-1 范围）
+    mean_values=[[123.675, 116.28, 103.53]],  # 默认输入是255, 使用这些值归一化到0-1
+    std_values=[[58.395, 57.12, 57.375]],
     quant_img_RGB2BGR=True,
     optimization_level=3,          # 最高优化级别
 )
 
 # 加载 ONNX
-ret = rknn.load_onnx(model="/home/admin-root/haotian/rk3588/pytorch模型转rknn/models/yolov8m.onnx")
+ret = rknn.load_onnx(model="/home/admin-root/haotian/康达瑞贝斯机器狗/det_shape.onnx")
 assert ret == 0, "加载 ONNX 失败！"
 
 # 转换模型
@@ -27,7 +27,7 @@ ret = rknn.build(
 assert ret == 0, "转换 RKNN 失败！"
 
 # 导出 RKNN
-ret = rknn.export_rknn("/home/admin-root/haotian/rk3588/pytorch模型转rknn/models/yolov8m.rknn")
+ret = rknn.export_rknn("/home/admin-root/haotian/康达瑞贝斯机器狗/det_shape.rknn")
 assert ret == 0, "导出 RKNN 失败！"