From 9d6ff65e1d8b42814c69e5955ec43ef5654f7ca9 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Thu, 16 Apr 2026 04:49:07 +0200 Subject: [PATCH] =?UTF-8?q?get=5Ftables=5Ffrom=5Fmodel:=20utilise=20artifi?= =?UTF-8?q?cial=20bound=20thresholding=E2=80=A6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit (to improve separation of neighbouring tables, esp. across columns; since model's threshold class is particularly weak, also use lower threshold here) --- src/eynollah/eynollah.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/eynollah/eynollah.py b/src/eynollah/eynollah.py index 9f118a7..d564f87 100644 --- a/src/eynollah/eynollah.py +++ b/src/eynollah/eynollah.py @@ -1539,7 +1539,11 @@ class Eynollah: return image_revised_last def get_tables_from_model(self, img): - prediction_table, _ = self.do_prediction_new_concept(False, img, self.model_zoo.get("table")) + prediction_table, _ = self.do_prediction_new_concept(False, img, + self.model_zoo.get("table"), + thresholding_for_artificial_class=True, + threshold_art_class=0.05, + artificial_class=2) prediction_table = prediction_table.astype(np.uint8) return prediction_table