add explicit ensure_dtype op

bioimage-io · Dec 10, 2024 · 6cc7343 · 6cc7343
1 parent 7d56157
commit 6cc7343
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/example_descriptions/models/unet2d_nuclei_broad/bioimageio.yaml b/example_descriptions/models/unet2d_nuclei_broad/bioimageio.yaml
@@ -95,6 +95,9 @@ outputs:
       sha256: 7bce8b53bcd0a12487a61f953aafe0f3700652848980d1083964c5bcb9555eec
     postprocessing:
       - id: sigmoid
+      - id: ensure_dtype
+        kwargs:
+          dtype: float32
 
 weights:
   pytorch_state_dict: