qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-04-30 19:22:03 +02:00
Code Issues Projects Releases Wiki Activity

add diagnostic plotting for prediction masking (commented)

Browse source
This commit is contained in:
Robert Sachunsky 2026-04-30 00:49:18 +02:00
parent 33c055389d
commit cbb3be0e01
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/eynollah/utils/__init__.py
View file

@ -1928,6 +1928,10 @@ def seg_mask_label(segmap:np.ndarray,
""" """
if not mask.any(): if not mask.any():
return return
# plt.subplot(2, 2, 1, title="segmap orig")
# plt.imshow(segmap)
# plt.subplot(2, 2, 2, title="mask")
# plt.imshow(mask)
if keep: if keep:
keepmask = segmap == keep keepmask = segmap == keep
if only: if only:
@ -1941,5 +1945,10 @@ def seg_mask_label(segmap:np.ndarray,
kernel = np.ones((dilate, dilate), np.uint8) kernel = np.ones((dilate, dilate), np.uint8)
mask = cv2.dilate(mask.astype(np.uint8), kernel, iterations=1) > 0 mask = cv2.dilate(mask.astype(np.uint8), kernel, iterations=1) > 0
segmap[mask] = label segmap[mask] = label
# plt.subplot(2, 2, 3, title="segmap masked")
# plt.imshow(segmap)
if keep: if keep:
segmap[keepmask] = keep segmap[keepmask] = keep
# plt.subplot(2, 2, 4, title="segmap final")
# plt.imshow(segmap)
# plt.show()