fix: crashing bug in CLI for contrast normalization

seung-lab · Dec 11, 2023 · 21f2f7b · 21f2f7b
1 parent c48be20
commit 21f2f7b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/igneous/task_creation/image.py b/igneous/task_creation/image.py
@@ -1120,11 +1120,13 @@ def create_contrast_normalization_tasks(
     dvol.info['scales'] = dvol.info['scales'][:mip+1]
     dvol.commit_info()
 
+  dvol.meta.unlock_mips(mip)
+
   if bounds is None:
     bounds = srcvol.bounds.clone()
 
   if shape is None:
-    shape = Bbox( (0,0,0), (2048, 2048, 64) )
+    shape = Bbox( (0,0,0), (2048, 2048, dvol.chunk_size.z) )
     shape = shape.shrink_to_chunk_size(dvol.underlying).size3()
     shape = Vec.clamp(shape, (1,1,1), bounds.size3() )
 

diff --git a/igneous_cli/cli.py b/igneous_cli/cli.py
@@ -528,7 +528,7 @@ def equalize(
   """(2) Apply histogram equalization to z-slices."""
   if bounds_mip is None:
     bounds_mip = mip
-  bounds = compute_bounds(path, bounds_mip, xrange, yrange, zrange)
+  bounds = compute_bounds(src, bounds_mip, xrange, yrange, zrange)
 
   tasks = tc.create_contrast_normalization_tasks(
     src, dest, levels_path=None,