refactor: class rename

This commit is contained in:
283375 2023-10-01 03:07:20 +08:00
parent c6aba3a7e9
commit 897705d23d
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
13 changed files with 26 additions and 26 deletions

View File

@ -1,2 +1,2 @@
from .auto import AutoMasker, AutoMaskerT1, AutoMaskerT2
from .common import Masker
from .auto import *
from .common import DeviceRoiMasker

View File

@ -1,3 +1,3 @@
from .common import AutoMasker
from .t1 import AutoMaskerT1
from .t2 import AutoMaskerT2
from .common import DeviceAutoRoiMasker
from .t1 import DeviceAutoRoiMaskerT1
from .t2 import DeviceAutoRoiMaskerT2

View File

@ -1,5 +1,5 @@
from ..common import Masker
from ..common import DeviceRoiMasker
class AutoMasker(Masker):
class DeviceAutoRoiMasker(DeviceRoiMasker):
...

View File

@ -1,7 +1,7 @@
import cv2
import numpy as np
from .common import AutoMasker
from .common import DeviceAutoRoiMasker
GRAY_BGR_MIN = np.array([50] * 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)
class AutoMaskerT1(AutoMasker):
class DeviceAutoRoiMaskerT1(DeviceAutoRoiMasker):
@classmethod
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

View File

@ -1,7 +1,7 @@
import cv2
import numpy as np
from .common import AutoMasker
from .common import DeviceAutoRoiMasker
PFL_HSV_MIN = np.array([0, 0, 248], 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)
class AutoMaskerT2(AutoMasker):
class DeviceAutoRoiMaskerT2(DeviceAutoRoiMasker):
@classmethod
def pfl(cls, roi_bgr: cv2.Mat) -> cv2.Mat:
return cv2.inRange(

View File

@ -1,7 +1,7 @@
import cv2
class Masker:
class DeviceRoiMasker:
@classmethod
def pure(cls, roi_bgr: cv2.Mat) -> cv2.Mat:
raise NotImplementedError()

View File

@ -1,2 +1,2 @@
from .common import Extractor
from .common import DeviceRoiExtractor
from .sizes import *

View File

@ -1,11 +1,11 @@
import cv2
from ..crop import crop_xywh
from .sizes.common import Sizes
from .sizes.common import DeviceRoiSizes
class Extractor:
def __init__(self, img: cv2.Mat, sizes: Sizes):
class DeviceRoiExtractor:
def __init__(self, img: cv2.Mat, sizes: DeviceRoiSizes):
self.img = img
self.sizes = sizes

View File

@ -1,3 +1,3 @@
from .common import AutoSizes
from .t1 import AutoSizesT1
from .t2 import AutoSizesT2
from .common import DeviceAutoRoiSizes
from .t1 import DeviceAutoRoiSizesT1
from .t2 import DeviceAutoRoiSizesT2

View File

@ -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):
self.w = w
self.h = h

View File

@ -1,7 +1,7 @@
from .common import AutoSizes
from .common import DeviceAutoRoiSizes
class AutoSizesT1(AutoSizes):
class DeviceAutoRoiSizesT1(DeviceAutoRoiSizes):
@property
def factor(self):
return (

View File

@ -1,7 +1,7 @@
from .common import AutoSizes
from .common import DeviceAutoRoiSizes
class AutoSizesT2(AutoSizes):
class DeviceAutoRoiSizesT2(DeviceAutoRoiSizes):
@property
def factor(self):
return (

View File

@ -3,7 +3,7 @@ from typing import Tuple
Rect = Tuple[int, int, int, int]
class Sizes:
class DeviceRoiSizes:
pure: Rect
far: Rect
lost: Rect