fix: cv2.Mat typing annotations

2025-07-05 14:26:27 +00:00 · 2023-10-23 13:29:17 +08:00
parent cf46bf7a59
commit f2f854040f
10 changed files with 85 additions and 76 deletions
--- a/src/arcaea_offline_ocr/device/rois/extractor/common.py
+++ b/src/arcaea_offline_ocr/device/rois/extractor/common.py
@ -1,11 +1,10 @@
-import cv2
-
 from ....crop import crop_xywh
+from ....types import Mat
 from ..definition.common import DeviceRois


 class DeviceRoisExtractor:
-    def __init__(self, img: cv2.Mat, rois: DeviceRois):
+    def __init__(self, img: Mat, rois: DeviceRois):
        self.img = img
        self.sizes = rois