From ae2d29ab15673efbf07f407c7a672b3376283782 Mon Sep 17 00:00:00 2001 From: Gregory Lee Date: Mon, 3 Mar 2025 10:36:03 -0500 Subject: [PATCH] update feret_diameter_max output dtype --- .../cucim/skimage/measure/_regionprops_gpu_convex.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/python/cucim/src/cucim/skimage/measure/_regionprops_gpu_convex.py b/python/cucim/src/cucim/skimage/measure/_regionprops_gpu_convex.py index aa95d7e8..890f0a01 100644 --- a/python/cucim/src/cucim/skimage/measure/_regionprops_gpu_convex.py +++ b/python/cucim/src/cucim/skimage/measure/_regionprops_gpu_convex.py @@ -92,15 +92,16 @@ def _feret_diameter_max(image_convex, spacing=None, return_argmax=False): spacing = cp.asarray(spacing, dtype=cp.float32).reshape(1, -1) coords *= spacing - out = pdist_max_blockwise( + squared_dist, index_argmax = pdist_max_blockwise( coords, metric="sqeuclidean", compute_argmax=return_argmax, coords_per_block=4000, ) + max_diameter = math.sqrt(float(squared_dist)) if return_argmax: - return math.sqrt(out[0]), out[1] - return math.sqrt(out[0]) + return max_diameter, index_argmax + return max_diameter def regionprops_feret_diameter_max( @@ -140,7 +141,8 @@ def regionprops_feret_diameter_max( image_convex, spacing=spacing, return_argmax=False ) for image_convex in images_convex - ) + ), + dtype=cp.float64, ) if props_dict is not None: props_dict["feret_diameter_max"] = diameters