From f3649adbf24eb6b4d189846d67eeed88f153ea06 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Thu, 21 May 2026 17:23:11 +0200 Subject: [PATCH] trocr: apply `do_not_mask_with_textline_contour` here, too --- src/eynollah/eynollah_ocr.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/eynollah/eynollah_ocr.py b/src/eynollah/eynollah_ocr.py index 747d2f5..f1b155b 100644 --- a/src/eynollah/eynollah_ocr.py +++ b/src/eynollah/eynollah_ocr.py @@ -113,9 +113,10 @@ class Eynollah_ocr(Eynollah): total_bb_coordinates.append([x, y, w, h]) img_crop = img[y: y + h, x: x + w] - mask_poly = np.zeros(img_crop.shape[:2], dtype=np.uint8) - mask_poly = cv2.fillPoly(mask_poly, pts=[cont - [x, y]], color=1) - img_crop[mask_poly == 0] = 255 # FIXME: or median color? + if not self.do_not_mask_with_textline_contour: + mask_poly = np.zeros(img_crop.shape[:2], dtype=np.uint8) + mask_poly = cv2.fillPoly(mask_poly, pts=[cont - [x, y]], color=1) + img_crop[mask_poly == 0] = 255 # FIXME: or median color? if h > 0.1 * w: cropped_lines.append(resize_image(img_crop,