mirror of
https://github.com/283375/arcaea-offline-ocr.git
synced 2025-04-18 13:00:18 +00:00
refactor: class rename
This commit is contained in:
parent
c6aba3a7e9
commit
897705d23d
@ -1,2 +1,2 @@
|
|||||||
from .auto import AutoMasker, AutoMaskerT1, AutoMaskerT2
|
from .auto import *
|
||||||
from .common import Masker
|
from .common import DeviceRoiMasker
|
||||||
|
@ -1,3 +1,3 @@
|
|||||||
from .common import AutoMasker
|
from .common import DeviceAutoRoiMasker
|
||||||
from .t1 import AutoMaskerT1
|
from .t1 import DeviceAutoRoiMaskerT1
|
||||||
from .t2 import AutoMaskerT2
|
from .t2 import DeviceAutoRoiMaskerT2
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
from ..common import Masker
|
from ..common import DeviceRoiMasker
|
||||||
|
|
||||||
|
|
||||||
class AutoMasker(Masker):
|
class DeviceAutoRoiMasker(DeviceRoiMasker):
|
||||||
...
|
...
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
import cv2
|
import cv2
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
from .common import AutoMasker
|
from .common import DeviceAutoRoiMasker
|
||||||
|
|
||||||
GRAY_BGR_MIN = np.array([50] * 3, np.uint8)
|
GRAY_BGR_MIN = np.array([50] * 3, np.uint8)
|
||||||
GRAY_BGR_MAX = np.array([160] * 3, np.uint8)
|
GRAY_BGR_MAX = np.array([160] * 3, np.uint8)
|
||||||
@ -34,7 +34,7 @@ PURE_MEMORY_HSV_MIN = np.array([90, 70, 80], np.uint8)
|
|||||||
PURE_MEMORY_HSV_MAX = np.array([110, 200, 175], np.uint8)
|
PURE_MEMORY_HSV_MAX = np.array([110, 200, 175], np.uint8)
|
||||||
|
|
||||||
|
|
||||||
class AutoMaskerT1(AutoMasker):
|
class DeviceAutoRoiMaskerT1(DeviceAutoRoiMasker):
|
||||||
@classmethod
|
@classmethod
|
||||||
def gray(cls, roi_bgr: cv2.Mat) -> cv2.Mat:
|
def gray(cls, roi_bgr: cv2.Mat) -> cv2.Mat:
|
||||||
bgr_value_equal_mask = np.max(roi_bgr, axis=2) - np.min(roi_bgr, axis=2) <= 5
|
bgr_value_equal_mask = np.max(roi_bgr, axis=2) - np.min(roi_bgr, axis=2) <= 5
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
import cv2
|
import cv2
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
from .common import AutoMasker
|
from .common import DeviceAutoRoiMasker
|
||||||
|
|
||||||
PFL_HSV_MIN = np.array([0, 0, 248], np.uint8)
|
PFL_HSV_MIN = np.array([0, 0, 248], np.uint8)
|
||||||
PFL_HSV_MAX = np.array([179, 10, 255], np.uint8)
|
PFL_HSV_MAX = np.array([179, 10, 255], np.uint8)
|
||||||
@ -38,7 +38,7 @@ PURE_MEMORY_HSV_MIN = np.array([90, 70, 80], np.uint8)
|
|||||||
PURE_MEMORY_HSV_MAX = np.array([110, 200, 175], np.uint8)
|
PURE_MEMORY_HSV_MAX = np.array([110, 200, 175], np.uint8)
|
||||||
|
|
||||||
|
|
||||||
class AutoMaskerT2(AutoMasker):
|
class DeviceAutoRoiMaskerT2(DeviceAutoRoiMasker):
|
||||||
@classmethod
|
@classmethod
|
||||||
def pfl(cls, roi_bgr: cv2.Mat) -> cv2.Mat:
|
def pfl(cls, roi_bgr: cv2.Mat) -> cv2.Mat:
|
||||||
return cv2.inRange(
|
return cv2.inRange(
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
import cv2
|
import cv2
|
||||||
|
|
||||||
|
|
||||||
class Masker:
|
class DeviceRoiMasker:
|
||||||
@classmethod
|
@classmethod
|
||||||
def pure(cls, roi_bgr: cv2.Mat) -> cv2.Mat:
|
def pure(cls, roi_bgr: cv2.Mat) -> cv2.Mat:
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
@ -1,2 +1,2 @@
|
|||||||
from .common import Extractor
|
from .common import DeviceRoiExtractor
|
||||||
from .sizes import *
|
from .sizes import *
|
||||||
|
@ -1,11 +1,11 @@
|
|||||||
import cv2
|
import cv2
|
||||||
|
|
||||||
from ..crop import crop_xywh
|
from ..crop import crop_xywh
|
||||||
from .sizes.common import Sizes
|
from .sizes.common import DeviceRoiSizes
|
||||||
|
|
||||||
|
|
||||||
class Extractor:
|
class DeviceRoiExtractor:
|
||||||
def __init__(self, img: cv2.Mat, sizes: Sizes):
|
def __init__(self, img: cv2.Mat, sizes: DeviceRoiSizes):
|
||||||
self.img = img
|
self.img = img
|
||||||
self.sizes = sizes
|
self.sizes = sizes
|
||||||
|
|
||||||
|
@ -1,3 +1,3 @@
|
|||||||
from .common import AutoSizes
|
from .common import DeviceAutoRoiSizes
|
||||||
from .t1 import AutoSizesT1
|
from .t1 import DeviceAutoRoiSizesT1
|
||||||
from .t2 import AutoSizesT2
|
from .t2 import DeviceAutoRoiSizesT2
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
from ..common import Sizes
|
from ..common import DeviceRoiSizes
|
||||||
|
|
||||||
|
|
||||||
class AutoSizes(Sizes):
|
class DeviceAutoRoiSizes(DeviceRoiSizes):
|
||||||
def __init__(self, w: int, h: int):
|
def __init__(self, w: int, h: int):
|
||||||
self.w = w
|
self.w = w
|
||||||
self.h = h
|
self.h = h
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
from .common import AutoSizes
|
from .common import DeviceAutoRoiSizes
|
||||||
|
|
||||||
|
|
||||||
class AutoSizesT1(AutoSizes):
|
class DeviceAutoRoiSizesT1(DeviceAutoRoiSizes):
|
||||||
@property
|
@property
|
||||||
def factor(self):
|
def factor(self):
|
||||||
return (
|
return (
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
from .common import AutoSizes
|
from .common import DeviceAutoRoiSizes
|
||||||
|
|
||||||
|
|
||||||
class AutoSizesT2(AutoSizes):
|
class DeviceAutoRoiSizesT2(DeviceAutoRoiSizes):
|
||||||
@property
|
@property
|
||||||
def factor(self):
|
def factor(self):
|
||||||
return (
|
return (
|
||||||
|
@ -3,7 +3,7 @@ from typing import Tuple
|
|||||||
Rect = Tuple[int, int, int, int]
|
Rect = Tuple[int, int, int, int]
|
||||||
|
|
||||||
|
|
||||||
class Sizes:
|
class DeviceRoiSizes:
|
||||||
pure: Rect
|
pure: Rect
|
||||||
far: Rect
|
far: Rect
|
||||||
lost: Rect
|
lost: Rect
|
||||||
|
Loading…
x
Reference in New Issue
Block a user