Commit f4b36fed authored by dmMaze's avatar dmMaze
Browse files

fix ocr text block list

parent 7516464f
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
from typing import Tuple, List, Dict, Union, Callable
from ordered_set import OrderedSet
import numpy as np
import cv2
import logging

from ..textdetector.textblock import TextBlock
@@ -36,6 +35,8 @@ class OCRBase(BaseModule):
        elif isinstance(blk_list, TextBlock):
            blk_list = [blk_list]

        for blk in blk_list:
            blk.text = []
        self.ocr_blk_list(img, blk_list)
        for blk in blk_list:
            if isinstance(blk.text, List):
+0 −1
Original line number Diff line number Diff line
@@ -51,7 +51,6 @@ class TextBlock:
    default_stroke_width: float = 0.2
    stroke_decide_by_colordiff: bool = True
    font_weight: int = 50
    _target_lang: str = ""
    opacity: float = 1.
    shadow_radius: float = 0.
    shadow_strength: float = 1.
+1 −1
Original line number Diff line number Diff line
@@ -855,7 +855,7 @@ class MainWindow(FramelessWindow):

        blk_list, blk_ids = [], []
        for blkitem in blkitem_list:
            blk = blkitem.blk
            blk: TextBlock = blkitem.blk
            blk._bounding_rect = blkitem.absBoundingRect()
            blk.vertical = blkitem.is_vertical
            blk.text = self.st_manager.pairwidget_list[blkitem.idx].e_source.toPlainText()