improv CTC solution compression (#17)

royerlab · Jan 9, 2023 · dca847f · dca847f
1 parent 26c5dfd
commit dca847f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ultrack/core/export/ctc.py b/ultrack/core/export/ctc.py
@@ -282,6 +282,9 @@ def _write_tiff_buffer(
         dilated = grey_dilation(buffer, footprint=footprint)
         np.putmask(buffer, buffer == 0, dilated)
 
+    # reducing mask sizes
+    buffer = buffer.astype(np.min_scalar_type(buffer.max()))
+
     imwrite(output_dir / f"mask{t:03}.tif", buffer)