Commit 4655a99a authored by dmMaze's avatar dmMaze
Browse files

fix textlines sorting in split_textblk

parent c9460d58

ballontranslator/dl/textdetector/textblock.py

+4 −2
Original line number Diff line number Diff line
@@ -409,7 +409,9 @@ def merge_textlines(blk_list: List[TextBlock]) -> List[TextBlock]: 
    return merged_list



def split_textblk(blk: TextBlock):

    font_size, distance, lines = blk.font_size, blk.distance, blk.lines_array()

    font_size, distance, lines = blk.font_size, blk.distance, blk.lines

    l0 = np.array(blk.lines[0])

    lines.sort(key=lambda line: np.linalg.norm(np.array(line[0] - l0[0])))

    distance_tol = font_size * 2

    current_blk = copy.deepcopy(blk)

    current_blk.lines = [lines[0]]
@@ -423,7 +425,7 @@ def split_textblk(blk: TextBlock): 
            if line_disance > distance_tol:

                split = True

            else:

                if blk.vertical and abs(abs(blk.angle) - 90) < 10:

                if blk.vertical and abs(blk.angle) < 10 and len(current_blk.lines) > 1:

                    split = abs(lines[jj][0][1] - line[0][1]) > font_size

        if split:

            current_blk = copy.deepcopy(current_blk)

ballontranslator/ui/dl_manager.py

+4 −1
Original line number Diff line number Diff line
@@ -289,7 +289,10 @@ class ImgtransThread(QThread): 
        self.inpaint_counter = 0

        self.num_pages = num_pages = len(self.imgtrans_proj.pages)



        if self.translator is not None:

            self.parallel_trans = not self.translator.is_computational_intensive()

        else:

            self.parallel_trans = False

        if self.dl_config.enable_translate and self.parallel_trans:

            self.translate_thread.runTranslatePipeline(self.imgtrans_proj)

ballontranslator/ui/canvas.py

+1 −1

File changed.

Contains only whitespace changes.