Expand context coordiante

brianreicher · Nov 14, 2023 · b37773e · b37773e
1 parent 64e71b9
commit b37773e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/rusty_mws/rusty_mws_segmentation.py b/src/rusty_mws/rusty_mws_segmentation.py
@@ -218,9 +218,8 @@ def __init__(
         if context is not None:
             self.context: Coordinate = context
         else:
-            self.context: Coordinate = Coordinate(
-                np.max(a=np.abs(neighborhood[:neighborhood_length]), axis=0)
-            )
+            affs_shape_half: list = [sz//2 for sz in open_ds(affs_file, affs_dataset, mode="r").data.shape[1:]]
+            self.context: Coordinate = Coordinate(affs_shape_half)
 
         self.filter_val: float = filter_val
         self.seeded: bool = seeded
@@ -324,7 +323,7 @@ def segment_seed_correction(
                 neighborhood_length=self.neighborhood_length,
                 mongo_port=self.mongo_port,
                 db_name=self.db_name,
-            )
+           )
 
         success = success & skel_correct_segmentation(
             seeds_file=self.seeds_file,