Commit b607a208 authored by dmMaze's avatar dmMaze
Browse files

fix translation direction for ysgyolo

parent 1ba9ef9d

modules/textdetector/detector_ysg.py

+1 −1
Original line number Diff line number Diff line
@@ -166,7 +166,7 @@ class YSGYoloDetector(TextDetectorBase): 
                            y2 += y1

                            cv2.rectangle(mask, (x1, y1), (x2, y2), 255, -1)

                        lines = xywh2xyxypoly(lines).reshape(-1, 4, 2).tolist()

                        blk = TextBlock(xyxy=xyxy, lines=np.array(lines), src_is_vertical=is_vertical)

                        blk = TextBlock(xyxy=xyxy, lines=np.array(lines), src_is_vertical=is_vertical, vertical=is_vertical)

                        blk.font_size = font_sz

                        blk._detected_font_size = font_sz

                        if is_vertical:

utils/textblock.py

+1 −1
Original line number Diff line number Diff line
@@ -898,7 +898,7 @@ def mit_merge_textlines(textlines: List[Quadrilateral], width: int, height: int, 
        is_vertical = nv >= len(txtlns) // 2

        region = TextBlock(

            lines=lines, text=texts, angle=angle, fontformat=ffmt, 

            _detected_font_size=font_size, src_is_vertical=is_vertical)

            _detected_font_size=font_size, src_is_vertical=is_vertical, vertical=is_vertical)

        region.adjust_bbox()

        if region.src_is_vertical:

            region.alignment = 1