283375/arcaea-offline-pyside-ui
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-pyside-ui.git synced 2025-04-19 09:10:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: crop black edges before ocr

Browse Source
This commit is contained in:
283375 2023-10-22 01:57:18 +08:00
parent e0d92b7784
commit 381f27db87
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ui/extends/tabs/tabOcr/tabOcr_Device.py
View File

@ -7,6 +7,7 @@ import exif
from arcaea_offline.database import Database from arcaea_offline.database import Database
from arcaea_offline.models import Chart, Score from arcaea_offline.models import Chart, Score
from arcaea_offline.utils.partner import KanaeDayNight, kanae_day_night from arcaea_offline.utils.partner import KanaeDayNight, kanae_day_night
from arcaea_offline_ocr.crop import CropBlackEdges
from arcaea_offline_ocr.device import DeviceOcr, DeviceOcrResult from arcaea_offline_ocr.device import DeviceOcr, DeviceOcrResult
from arcaea_offline_ocr.device.rois import ( from arcaea_offline_ocr.device.rois import (
DeviceRois, DeviceRois,
@ -43,6 +44,7 @@ class TabDeviceOcrRunnable(OcrRunnable):
def run(self): def run(self):
try: try:
img = imread_unicode(self.imagePath, cv2.IMREAD_COLOR) img = imread_unicode(self.imagePath, cv2.IMREAD_COLOR)
img = CropBlackEdges.crop(img, cv2.COLOR_BGR2GRAY)
if isinstance(self.rois, type) and issubclass(self.rois, DeviceRoisAuto): if isinstance(self.rois, type) and issubclass(self.rois, DeviceRoisAuto):
rois = self.rois(img.shape[1], img.shape[0]) rois = self.rois(img.shape[1], img.shape[0])
else: else: