From 151b1fd6d866077253d7c32ead3c17c1d771a25c Mon Sep 17 00:00:00 2001 From: 283375 Date: Fri, 29 Sep 2023 02:26:17 +0800 Subject: [PATCH] refactor: basic structure --- src/arcaea_offline_ocr/extractor/__init__.py | 0 src/arcaea_offline_ocr/extractor/common.py | 5 ++ .../extractor/sizes/__init__.py | 0 .../extractor/sizes/common.py | 15 +++++ src/arcaea_offline_ocr/masker/__init__.py | 0 src/arcaea_offline_ocr/masker/common.py | 55 +++++++++++++++++++ 6 files changed, 75 insertions(+) create mode 100644 src/arcaea_offline_ocr/extractor/__init__.py create mode 100644 src/arcaea_offline_ocr/extractor/common.py create mode 100644 src/arcaea_offline_ocr/extractor/sizes/__init__.py create mode 100644 src/arcaea_offline_ocr/extractor/sizes/common.py create mode 100644 src/arcaea_offline_ocr/masker/__init__.py create mode 100644 src/arcaea_offline_ocr/masker/common.py diff --git a/src/arcaea_offline_ocr/extractor/__init__.py b/src/arcaea_offline_ocr/extractor/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/arcaea_offline_ocr/extractor/common.py b/src/arcaea_offline_ocr/extractor/common.py new file mode 100644 index 0000000..64d5e49 --- /dev/null +++ b/src/arcaea_offline_ocr/extractor/common.py @@ -0,0 +1,5 @@ +from .sizes.common import Sizes + + +class Extractor: + sizes: Sizes diff --git a/src/arcaea_offline_ocr/extractor/sizes/__init__.py b/src/arcaea_offline_ocr/extractor/sizes/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/arcaea_offline_ocr/extractor/sizes/common.py b/src/arcaea_offline_ocr/extractor/sizes/common.py new file mode 100644 index 0000000..fa85a20 --- /dev/null +++ b/src/arcaea_offline_ocr/extractor/sizes/common.py @@ -0,0 +1,15 @@ +from typing import Tuple + +Rect = Tuple[int, int, int, int] + + +class Sizes: + pure: Rect + far: Rect + lost: Rect + score: Rect + rating_class: Rect + max_recall: Rect + jacket: Rect + clear_status: Rect + partner_icon: Rect diff --git a/src/arcaea_offline_ocr/masker/__init__.py b/src/arcaea_offline_ocr/masker/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/arcaea_offline_ocr/masker/common.py b/src/arcaea_offline_ocr/masker/common.py new file mode 100644 index 0000000..8b8c867 --- /dev/null +++ b/src/arcaea_offline_ocr/masker/common.py @@ -0,0 +1,55 @@ +import cv2 + + +class Masker: + @staticmethod + def pure(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def far(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def lost(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def score(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def rating_class_pst(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def rating_class_prs(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def rating_class_ftr(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def rating_class_byd(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def max_recall(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def clear_status_track_lost(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def clear_status_track_complete(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def clear_status_full_recall(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError() + + @staticmethod + def clear_status_pure_memory(roi_bgr: cv2.Mat) -> cv2.Mat: + raise NotImplementedError()