add low_threshold dtype fix in _nonmaximum_suppression kernel for canny

rapidsai · Jun 19, 2023 · 4e246bb · 4e246bb
1 parent 91ad7ef
commit 4e246bb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/cucim/src/cucim/skimage/feature/_canny.py b/python/cucim/src/cucim/skimage/feature/_canny.py
@@ -235,6 +235,11 @@ def _nonmaximum_suppression_bilinear(
     kernel = _get_nonmax_kernel(large_int=large_int)
 
     out = cp.empty_like(magnitude)
+
+    if isinstance(low_threshold, cp.ndarray):
+        # if array scalar was provided, make sure dtype matches other arrays
+        low_threshold = low_threshold.astype(out.dtype, copy=False)
+
     kernel(isobel, jsobel, magnitude, eroded_mask, low_threshold, out)
     return out