haotian/convert-the-model-to-rknn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sss

Browse Source
This commit is contained in:
haotian 2025-08-19 16:32:01 +08:00
parent af901054af
commit 3ef17b11f1
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
009PaddleOCR_ONNX转rknn.py
View File

@ -9,8 +9,10 @@ rknn = RKNN()
# 配置参数（关键！） # 配置参数（关键！）
rknn.config( rknn.config(
target_platform="rk3588", # 根据实际芯片型号修改 target_platform="rk3588", # 根据实际芯片型号修改
mean_values=[[123.675, 116.28, 103.53]], # 默认输入是255, 使用这些值归一化到0-1 # mean_values=[[123.675, 116.28, 103.53]], # 默认输入是255, 使用这些值归一化到0-1
std_values=[[58.395, 57.12, 57.375]], # std_values=[[58.395, 57.12, 57.375]],
mean_values=[[0, 0, 0]], # 默认输入是255, 使用这些值归一化到0-1
std_values=[[255, 255, 255]],
quant_img_RGB2BGR=True, quant_img_RGB2BGR=True,
optimization_level=3, # 最高优化级别 optimization_level=3, # 最高优化级别
) )
@ -27,7 +29,7 @@ ret = rknn.build(
assert ret == 0, "转换 RKNN 失败！" assert ret == 0, "转换 RKNN 失败！"
# 导出 RKNN # 导出 RKNN
ret = rknn.export_rknn("/home/admin-root/haotian/康达瑞贝斯机器狗/rec_shape_20250815.rknn") ret = rknn.export_rknn("/home/admin-root/haotian/康达瑞贝斯机器狗/rec_shape_20250815_0m.rknn")
assert ret == 0, "导出 RKNN 失败！" assert ret == 0, "导出 RKNN 失败！"

BIN
tflite_mobilenet_v1_0.npy
View File

Binary file not shown.