qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-08-09 04:12:42 +02:00
Code Issues Projects Releases Wiki Activity

tiling for enhancement: ensure precision by rounding

Browse source
This commit is contained in:
Robert Sachunsky 2026-07-30 15:29:03 +02:00
parent b2777a2562
commit 5f4e4fb8bd
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/eynollah/utils/tiling.py
View file

@ -159,7 +159,7 @@ def do_prediction(
img_patch[:] = 0
if is_enhancement:
seg = (prediction * 255).astype(np.uint8)
seg = np.round(prediction * 255).astype(np.uint8)
else:
seg = np.argmax(prediction, axis=2).astype(np.uint8)
if thresholding_for_some_classes: