From 442fa2de7574bd36d8dbba3655ce623466e6aeca Mon Sep 17 00:00:00 2001 From: 283375 Date: Wed, 27 Sep 2023 23:57:23 +0800 Subject: [PATCH] fix: ignore image alpha channel --- ui/extends/tabs/tabOcr/tabOcr_Device.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/ui/extends/tabs/tabOcr/tabOcr_Device.py b/ui/extends/tabs/tabOcr/tabOcr_Device.py index 72ab204..795ee10 100644 --- a/ui/extends/tabs/tabOcr/tabOcr_Device.py +++ b/ui/extends/tabs/tabOcr/tabOcr_Device.py @@ -2,6 +2,7 @@ import contextlib import logging from typing import Tuple +import cv2 from arcaea_offline.database import Database from arcaea_offline.models import Chart, Score from arcaea_offline_ocr.device.shared import DeviceOcrResult @@ -28,7 +29,9 @@ class TabDeviceV2OcrRunnable(OcrRunnable): def run(self): try: - rois = DeviceV2Rois(self.device, imread_unicode(self.imagePath)) + rois = DeviceV2Rois( + self.device, imread_unicode(self.imagePath, cv2.IMREAD_COLOR) + ) rois.sizes = ( SizesV2(self.device.factor) if self.sizesV2 @@ -53,7 +56,7 @@ class TabDeviceV2AutoRoisOcrRunnable(OcrRunnable): def run(self): try: - rois = DeviceV2AutoRois(imread_unicode(self.imagePath)) + rois = DeviceV2AutoRois(imread_unicode(self.imagePath, cv2.IMREAD_COLOR)) factor = rois.sizes.factor rois.sizes = SizesV2(factor) if self.sizesV2 else SizesV1(factor) ocr = DeviceV2Ocr(self.knnModel, self.phashDb)