From ccbfd4f80d87bd433273b37b707dd151086e3d93 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Fri, 10 Nov 2023 16:34:50 +0100
Subject: [PATCH] Automatically export boolean masks as 0-255 black/white
---
model_server/accessors.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/model_server/accessors.py b/model_server/accessors.py
index 0e5e2cc9..62df189d 100644
--- a/model_server/accessors.py
+++ b/model_server/accessors.py
@@ -189,7 +189,11 @@ def write_accessor_data_to_file(fpath: Path, accessor: GenericImageDataAccessor,
[0, 1, 2, 3]
)
if accessor.is_mask():
- tifffile.imwrite(fpath, zcyx.astype('uint8'), imagej=True)
+ if accessor.dtype == 'bool':
+ data = zcyx * 255
+ else:
+ data = zcyx.astype('uint8')
+ tifffile.imwrite(fpath, data, imagej=True)
else:
tifffile.imwrite(fpath, zcyx, imagej=True)
except:
--
GitLab