diff --git a/arknights_mower/solvers/base_schedule.py b/arknights_mower/solvers/base_schedule.py index 0117f7c6e..74a9264ad 100644 --- 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: