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

1.实现016新的导出rknn模型脚本\n2.017实现使用rknn模型的脚本

Browse Source
This commit is contained in:
haotian 2025-08-26 15:44:21 +08:00
parent b3a781dbcf
commit 10761b756c
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
013PaddleOCR_ONNX文本检测模型转rknn.py
View File

@ -6,9 +6,12 @@ def convert_det_model():
# 配置模型 # 配置模型
print('--> Config model') print('--> Config model')
rknn.config(mean_values=[[123.675, 116.28, 103.53]], rknn.config(
mean_values=[[123.675, 116.28, 103.53]],
std_values=[[58.395, 57.12, 57.375]], std_values=[[58.395, 57.12, 57.375]],
quant_img_RGB2BGR=True, # mean_values=[[0, 0, 0]],
# std_values=[[255, 255, 255]],
# quant_img_RGB2BGR=True,
target_platform='rk3588') # 根据您的芯片型号调整 target_platform='rk3588') # 根据您的芯片型号调整
# 加载ONNX模型 # 加载ONNX模型
@ -28,7 +31,7 @@ def convert_det_model():
exit(ret) exit(ret)
# 导出RKNN模型 # 导出RKNN模型
ret = rknn.export_rknn('./det_shape_bgr.rknn') ret = rknn.export_rknn('./det_shape_255.rknn')
if ret != 0: if ret != 0:
print('Export model failed!') print('Export model failed!')
exit(ret) exit(ret)

4
014PaddleOCR_ONNX文本识别模型转rknn.py
View File

@ -7,7 +7,7 @@ def convert_rec_model():
print('--> Config model') print('--> Config model')
rknn.config(mean_values=[[127.5, 127.5, 127.5]], rknn.config(mean_values=[[127.5, 127.5, 127.5]],
std_values=[[127.5, 127.5, 127.5]], std_values=[[127.5, 127.5, 127.5]],
quant_img_RGB2BGR=True, # quant_img_RGB2BGR=True,
target_platform='rk3588') target_platform='rk3588')
print('--> Loading model') print('--> Loading model')
@ -24,7 +24,7 @@ def convert_rec_model():
print('Build model failed!') print('Build model failed!')
exit(ret) exit(ret)
ret = rknn.export_rknn('./rec_model_bgr.rknn') ret = rknn.export_rknn('./rec_model.rknn')
if ret != 0: if ret != 0:
print('Export model failed!') print('Export model failed!')
exit(ret) exit(ret)