Adapt layout model tresholds and remove blacklist

Signed-off-by: Christoph Auer <[email protected]>
DS4SD · Dec 4, 2024 · b83cd6c · b83cd6c
1 parent c73f914
commit b83cd6c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docling_ibm_models/layoutmodel/layout_predictor.py b/docling_ibm_models/layoutmodel/layout_predictor.py
@@ -63,10 +63,10 @@ def __init__(
         }
 
         # Blacklisted classes
-        self._black_classes = set(["Form", "Key-Value Region"])
+        self._black_classes = set()  # ["Form", "Key-Value Region"])
 
         # Set basic params
-        self._threshold = 0.6  # Score threshold
+        self._threshold = 0.5  # Score threshold
         self._image_size = 640
         self._size = np.asarray([[self._image_size, self._image_size]], dtype=np.int64)