diff --git a/013PaddleOCR_ONNX文本检测模型转rknn.py b/013PaddleOCR_ONNX文本检测模型转rknn.py index 6658455..1ed66ec 100644 --- a/013PaddleOCR_ONNX文本检测模型转rknn.py +++ b/013PaddleOCR_ONNX文本检测模型转rknn.py @@ -6,9 +6,12 @@ def convert_det_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]], - quant_img_RGB2BGR=True, + # mean_values=[[0, 0, 0]], + # std_values=[[255, 255, 255]], + # quant_img_RGB2BGR=True, target_platform='rk3588') # 根据您的芯片型号调整 # 加载ONNX模型 @@ -28,7 +31,7 @@ def convert_det_model(): exit(ret) # 导出RKNN模型 - ret = rknn.export_rknn('./det_shape_bgr.rknn') + ret = rknn.export_rknn('./det_shape_255.rknn') if ret != 0: print('Export model failed!') exit(ret) diff --git a/014PaddleOCR_ONNX文本识别模型转rknn.py b/014PaddleOCR_ONNX文本识别模型转rknn.py index 2e43282..e282935 100644 --- a/014PaddleOCR_ONNX文本识别模型转rknn.py +++ b/014PaddleOCR_ONNX文本识别模型转rknn.py @@ -7,7 +7,7 @@ def convert_rec_model(): print('--> Config model') rknn.config(mean_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') print('--> Loading model') @@ -24,7 +24,7 @@ def convert_rec_model(): print('Build model failed!') exit(ret) - ret = rknn.export_rknn('./rec_model_bgr.rknn') + ret = rknn.export_rknn('./rec_model.rknn') if ret != 0: print('Export model failed!') exit(ret)