mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
[Hackathon 182 Model] Update PPOCRV3 For RKNPU2 (#1403)
* update ppocrv3 for rknpu2 * add config * add config * detele unuseful * update useful results * Repair note * Repair note * fixed bugs * update
This commit is contained in:
@@ -0,0 +1,15 @@
|
||||
mean:
|
||||
-
|
||||
- 127.5
|
||||
- 127.5
|
||||
- 127.5
|
||||
std:
|
||||
-
|
||||
- 127.5
|
||||
- 127.5
|
||||
- 127.5
|
||||
model_path: ./ch_ppocr_mobile_v2.0_cls_infer/ch_ppocr_mobile_v2.0_cls_infer.onnx
|
||||
outputs_nodes:
|
||||
do_quantization: False
|
||||
dataset:
|
||||
output_folder: "./ch_ppocr_mobile_v2.0_cls_infer"
|
||||
@@ -0,0 +1,15 @@
|
||||
mean:
|
||||
-
|
||||
- 123.675
|
||||
- 116.28
|
||||
- 103.53
|
||||
std:
|
||||
-
|
||||
- 58.395
|
||||
- 57.12
|
||||
- 57.375
|
||||
model_path: ./ch_PP-OCRv3_det_infer/ch_PP-OCRv3_det_infer.onnx
|
||||
outputs_nodes:
|
||||
do_quantization: False
|
||||
dataset:
|
||||
output_folder: "./ch_PP-OCRv3_det_infer"
|
||||
@@ -0,0 +1,15 @@
|
||||
mean:
|
||||
-
|
||||
- 127.5
|
||||
- 127.5
|
||||
- 127.5
|
||||
std:
|
||||
-
|
||||
- 127.5
|
||||
- 127.5
|
||||
- 127.5
|
||||
model_path: ./ch_PP-OCRv3_rec_infer/ch_PP-OCRv3_rec_infer.onnx
|
||||
outputs_nodes:
|
||||
do_quantization: False
|
||||
dataset:
|
||||
output_folder: "./ch_PP-OCRv3_rec_infer"
|
||||
@@ -65,7 +65,10 @@ if __name__ == "__main__":
|
||||
if not os.path.exists(yaml_config["output_folder"]):
|
||||
os.mkdir(yaml_config["output_folder"])
|
||||
|
||||
model_base_name = os.path.basename(yaml_config["model_path"]).split(".")[0]
|
||||
name_list = os.path.basename(yaml_config["model_path"]).split(".")
|
||||
model_base_name = ""
|
||||
for name in name_list[0:-1]:
|
||||
model_base_name += name
|
||||
model_device_name = config.target_platform.lower()
|
||||
if yaml_config["do_quantization"]:
|
||||
model_save_name = model_base_name + "_" + model_device_name + "_quantized" + ".rknn"
|
||||
|
||||
Reference in New Issue
Block a user