From cbb3be0e014f8c792a1a599914cc34108b3c6ce9 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Thu, 30 Apr 2026 00:49:18 +0200 Subject: [PATCH] add diagnostic plotting for prediction masking (commented) --- src/eynollah/utils/__init__.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/eynollah/utils/__init__.py b/src/eynollah/utils/__init__.py index 7a4467d..47a765c 100644 --- a/src/eynollah/utils/__init__.py +++ b/src/eynollah/utils/__init__.py @@ -1928,6 +1928,10 @@ def seg_mask_label(segmap:np.ndarray, """ if not mask.any(): return + # plt.subplot(2, 2, 1, title="segmap orig") + # plt.imshow(segmap) + # plt.subplot(2, 2, 2, title="mask") + # plt.imshow(mask) if keep: keepmask = segmap == keep if only: @@ -1941,5 +1945,10 @@ def seg_mask_label(segmap:np.ndarray, kernel = np.ones((dilate, dilate), np.uint8) mask = cv2.dilate(mask.astype(np.uint8), kernel, iterations=1) > 0 segmap[mask] = label + # plt.subplot(2, 2, 3, title="segmap masked") + # plt.imshow(segmap) if keep: segmap[keepmask] = keep + # plt.subplot(2, 2, 4, title="segmap final") + # plt.imshow(segmap) + # plt.show()