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

fix: ignore image alpha channel

Browse Source
This commit is contained in:
283375 2023-09-27 23:57:23 +08:00
parent 3c80981c8a
commit 442fa2de75
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,6 +2,7 @@ import contextlib
import logging import logging
from typing import Tuple from typing import Tuple
import cv2
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_ocr.device.shared import DeviceOcrResult from arcaea_offline_ocr.device.shared import DeviceOcrResult
@ -28,7 +29,9 @@ class TabDeviceV2OcrRunnable(OcrRunnable):
def run(self): def run(self):
try: try:
rois = DeviceV2Rois(self.device, imread_unicode(self.imagePath)) rois = DeviceV2Rois(
self.device, imread_unicode(self.imagePath, cv2.IMREAD_COLOR)
)
rois.sizes = ( rois.sizes = (
SizesV2(self.device.factor) SizesV2(self.device.factor)
if self.sizesV2 if self.sizesV2
@ -53,7 +56,7 @@ class TabDeviceV2AutoRoisOcrRunnable(OcrRunnable):
def run(self): def run(self):
try: try:
rois = DeviceV2AutoRois(imread_unicode(self.imagePath)) rois = DeviceV2AutoRois(imread_unicode(self.imagePath, cv2.IMREAD_COLOR))
factor = rois.sizes.factor factor = rois.sizes.factor
rois.sizes = SizesV2(factor) if self.sizesV2 else SizesV1(factor) rois.sizes = SizesV2(factor) if self.sizesV2 else SizesV1(factor)
ocr = DeviceV2Ocr(self.knnModel, self.phashDb) ocr = DeviceV2Ocr(self.knnModel, self.phashDb)