OCR初始化推迟到调用

auto_get_res.py

@@ -15,6 +15,7 @@ from sklearn.neighbors import KNeighborsClassifier
 
 from mower.utils.character_recognize import fast_keypoints
 from mower.utils.image import loadimg, thres2
+from mower.utils.tile_pos import Level
 from mower.utils.vector import sa
 
 
@@ -677,8 +678,9 @@ class Arknights数据处理器:
 
     def levels(self):
         levels_json = 加载json("./ArknightsGameResource/levels.json")
+        LEVELS = [Level.from_json(data) for data in levels_json]
         with lzma.open("mower/static/levels.pkl", "wb") as f:
-            pickle.dump(levels_json, f)
+            pickle.dump(LEVELS, f)
 
     def 读取模组(self):
         _ = []

mower/utils/rapidocr.py

@@ -3,12 +3,14 @@ from rapidocr_onnxruntime import RapidOCR
 from mower.utils import typealias as tp
 from mower.utils.log import logger
 
-logger.info("加载OCR")
+engine = None
-engine = RapidOCR(text_score=0.3)
-logger.info("OCR加载完成")
 
 
 def ocr_rec(img: tp.Image | tp.GrayImage) -> str | None:
+    global engine
+    if engine is None:
+        engine = RapidOCR(text_score=0.3)
+
     result, elapse = engine(img, use_det=False, use_cls=False, use_rec=True)
     elapse = round(elapse[0] * 1000)
     try:

mower/utils/tile_pos.py

@@ -238,10 +238,8 @@ class Calc:
         return Vector2(x * self.screen_width, (1 - y) * self.screen_height)
 
 
-LEVELS: List[Level] = []
 with lzma.open(f"{__rootdir__}/static/levels.pkl", "rb") as f:
-    for data in pickle.load(f):
+    LEVELS: List[Level] = pickle.load(f)
-        LEVELS.append(Level.from_json(data))
 
 
 def find_level(code: Optional[str], name: Optional[str]) -> Optional[Level]: