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

chore(rapidocr_onnxruntime): Optim tips of infer_engine

Browse Source
This commit is contained in:
SWHL committed 2024-05-18 15:27:13 +08:00
1 parent de44841377
commit 7528fa9660
1 file changed
+7 -4
Download Patch File Download Diff File Expand all files Collapse all files
python/rapidocr_onnxruntime/utils/infer_engine.py
+7 -4
View File
@@ -96,7 +96,7 @@ class OrtInferSession:
return True
self.logger.warning(
"%s is not in available providers (%s), default use of %s inference.",
"%s is not in available providers (%s). Use %s inference by default.",
CUDA_EP,
self.had_providers,
self.had_providers[0],
@@ -127,15 +127,18 @@ class OrtInferSession:
cur_os = platform.system()
if cur_os != "Windows":
self.logger.warning(
"DirectML is only supported in Windows OS. The current OS is %s", cur_os
"DirectML is only supported in Windows OS. The current OS is %s. Use %s inference by default.",
cur_os,
self.had_providers[0],
)
return False
cur_window_version = int(platform.release().split(".")[0])
if cur_window_version < 10:
self.logger.warning(
"DirectML is only supported in Windows 10 and above OS. The current Windows version is %s",
"DirectML is only supported in Windows 10 and above OS. The current Windows version is %s. Use %s inference by default.",
cur_window_version,
self.had_providers[0],
)
return False
@@ -143,7 +146,7 @@ class OrtInferSession:
return True
self.logger.warning(
"%s is not in available providers (%s), default use of %s inference.",
"%s is not in available providers (%s). Use %s inference by default.",
DIRECTML_EP,
self.had_providers,
self.had_providers[0],