Fix bug in segmentation algorithm

NBCLab · Apr 19, 2023 · 70bf2e6 · 70bf2e6
1 parent 42ed691
commit 70bf2e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/segmentation.py b/segmentation.py
@@ -202,7 +202,7 @@ def _fit(self, gradient):
                 map_arr[labels_arr == i] = gradient[labels_arr == i]
                 gradient_maps.append(map_arr)
 
-                map_bounds.append(map_arr.max())
+                map_bounds.append(gradient[labels_arr == i].max())
 
             segments.append(gradient_maps)
             labels.append(labels_arr)