mtfyun/rapidocr
1
0 Fork 0
mirror of https://github.com/RapidAI/RapidOCR.git synced 2026-11-13 06:20:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(rapidocr_onnxruntime): Fix dml not work issue (#180)

Browse Source
This commit is contained in:
SkeathyTomas authored and GitHub committed 2024-05-20 21:06:39 +08:00
1 parent 3ad90cfce4
commit fb23f82870
1 file changed
+3 -3
Download Patch File Download Diff File Expand all files Collapse all files
python/rapidocr_onnxruntime/utils/parse_parameters.py
+3 -3
View File
@@ -149,15 +149,15 @@ class UpdateParameters:
global_dict, det_dict, cls_dict, rec_dict = self.parse_kwargs(**kwargs)
new_config = {
"Global": self.update_global_params(config["Global"], global_dict),
"Det": self.update_params(config["Det"], det_dict, "det_", None),
"Det": self.update_params(config["Det"], det_dict, "det_", ["det_model_path", "det_use_cuda", "det_use_dml"]),
"Cls": self.update_params(
config["Cls"],
cls_dict,
"cls_",
["cls_label_list", "cls_model_path", "cls_use_cuda"],
["cls_label_list", "cls_model_path", "cls_use_cuda", "cls_use_dml"],
),
"Rec": self.update_params(
config["Rec"], rec_dict, "rec_", ["rec_model_path", "rec_use_cuda"]
config["Rec"], rec_dict, "rec_", ["rec_model_path", "rec_use_cuda", "rec_use_dml"]
),
}