283375/arcaea-offline-ocr
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-ocr.git synced 2025-07-04 13:56:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: partial numpy & cv2 import to full import

Browse Source
This commit is contained in:
283375
2023-08-09 19:05:17 +08:00
parent 920ceffb35
commit 64598d0a84
7 changed files with 41 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/arcaea_offline_ocr/device/v1/crop.py
View File

@ -1,7 +1,7 @@
from math import floor
from typing import Any, Tuple
from numpy import all, array, count_nonzero
import numpy as np
from ...types import Mat
from .definition import Device
@ -59,6 +59,8 @@ def crop_to_title(screenshot: Mat, device: Device):
def is_black_edge(list_of_pixels: Mat, black_pixel=None):
if black_pixel is None:
black_pixel = array([0, 0, 0], list_of_pixels.dtype)
black_pixel = np.array([0, 0, 0], list_of_pixels.dtype)
pixels = list_of_pixels.reshape([-1, 3])
return count_nonzero(all(pixels < black_pixel, axis=1)) > floor(len(pixels) * 0.6)
return np.count_nonzero(all(pixels < black_pixel, axis=1)) > floor(
len(pixels) * 0.6
)

2
src/arcaea_offline_ocr/device/v2/shared.py
View File

@ -1,4 +1,4 @@
from cv2 import MORPH_CROSS, MORPH_ELLIPSE, MORPH_RECT, getStructuringElement
from cv2 import MORPH_RECT, getStructuringElement
PFL_DENOISE_KERNEL = getStructuringElement(MORPH_RECT, [2, 2])
PFL_ERODE_KERNEL = getStructuringElement(MORPH_RECT, [3, 3])