From 5dd5a1b0077be67f2b83670cf9583299f3ca9d8a Mon Sep 17 00:00:00 2001 From: felix Date: Mon, 4 Dec 2023 11:24:21 +0100 Subject: [PATCH] update --- doctr/models/detection/differentiable_binarization/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doctr/models/detection/differentiable_binarization/base.py b/doctr/models/detection/differentiable_binarization/base.py index af7cfa2d11..b28f75b7ac 100644 --- a/doctr/models/detection/differentiable_binarization/base.py +++ b/doctr/models/detection/differentiable_binarization/base.py @@ -299,7 +299,7 @@ def build_target( seg_target: np.ndarray = np.zeros(target_shape, dtype=np.uint8) seg_mask: np.ndarray = np.ones(target_shape, dtype=bool) thresh_target: np.ndarray = np.zeros(target_shape, dtype=np.float32) - thresh_mask: np.ndarray = np.ones(target_shape, dtype=np.uint8) + thresh_mask: np.ndarray = np.zeros(target_shape, dtype=np.uint8) for idx, tgt in enumerate(target): for class_idx, _tgt in enumerate(tgt.values()):