PaddleOCR模型保存至tmp中，兼容中文用户名

ArkMowers · Sep 27, 2023 · 091cff5 · 091cff5
1 parent 0fbd0f4
commit 091cff5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/arknights_mower/solvers/base_schedule.py b/arknights_mower/solvers/base_schedule.py
@@ -896,9 +896,12 @@ def double_read_time(self, cord, upperLimit=None, use_digit_reader=False):
         return execute_time
 
     def initialize_paddle(self):
+        det_model_dir = os.path.join(os.getcwd(), "tmp", "paddle", "det", "ch")
+        rec_model_dir = os.path.join(os.getcwd(), "tmp", "paddle", "rec", "ch")
+        cls_model_dir = os.path.join(os.getcwd(), "tmp", "paddle", "cls")
         global ocr
         if ocr is None:
-            ocr = PaddleOCR(enable_mkldnn=False, use_angle_cls=False, cls=False, show_log=False)
+            ocr = PaddleOCR(enable_mkldnn=False, use_angle_cls=False, cls=False, show_log=False, det_model_dir=det_model_dir, rec_model_dir=rec_model_dir, cls_model_dir=cls_model_dir)
 
     def read_screen(self, img, type="mood", limit=24, cord=None):
         if cord is not None: