From 1aa7b51b0bba87fa8a069dfbce665f8d3636b67d Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Mon, 13 Nov 2023 11:03:29 +0100
Subject: [PATCH] Merged in mask export method
---
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..4f122fb1 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).astype('uint8')
+ else:
+ data = zcyx.astype('uint8')
+ tifffile.imwrite(fpath, data, imagej=True)
else:
tifffile.imwrite(fpath, zcyx, imagej=True)
except:
--
GitLab