o :ήcr@s0ddlZddlZddlmZddZddZdS)N)cKDTreecCstt|}tt|}t|dkr!t|dkrdStjSt|dkr*tjSttt|j|dddtt|j|dddS)aCalculate the Hausdorff distance between nonzero elements of given images. The Hausdorff distance [1]_ is the maximum distance between any point on ``image0`` and its nearest point on ``image1``, and vice-versa. Parameters ---------- image0, image1 : ndarray Arrays where ``True`` represents a point that is included in a set of points. Both arrays must have the same shape. Returns ------- distance : float The Hausdorff distance between coordinates of nonzero pixels in ``image0`` and ``image1``, using the Euclidian distance. References ---------- .. [1] http://en.wikipedia.org/wiki/Hausdorff_distance Examples -------- >>> points_a = (3, 0) >>> points_b = (6, 0) >>> shape = (7, 1) >>> image_a = np.zeros(shape, dtype=bool) >>> image_b = np.zeros(shape, dtype=bool) >>> image_a[points_a] = True >>> image_b[points_b] = True >>> hausdorff_distance(image_a, image_b) 3.0 r)k)np transposenonzeroleninfmaxrquery)image0image1a_pointsb_pointsrB/tmp/pip-target-vg8gfxp4/lib/python/skimage/metrics/set_metrics.pyhausdorff_distances#  rc Cstt|}tt|}t|dkst|dkr%tjddddSt||\}}t||\}}|}|} ||} || } | | krU|| ||| fS|||||fS)aReturns pair of points that are Hausdorff distance apart between nonzero elements of given images. The Hausdorff distance [1]_ is the maximum distance between any point on ``image0`` and its nearest point on ``image1``, and vice-versa. Parameters ---------- image0, image1 : ndarray Arrays where ``True`` represents a point that is included in a set of points. Both arrays must have the same shape. Returns ------- point_a, point_b : array A pair of points that have Hausdorff distance between them. References ---------- .. [1] http://en.wikipedia.org/wiki/Hausdorff_distance Examples -------- >>> points_a = (3, 0) >>> points_b = (6, 0) >>> shape = (7, 1) >>> image_a = np.zeros(shape, dtype=bool) >>> image_b = np.zeros(shape, dtype=bool) >>> image_a[points_a] = True >>> image_b[points_b] = True >>> hausdorff_pair(image_a, image_b) (array([3, 0]), array([6, 0])) rz#One or both of the images is empty.) stacklevel)rr) rrrrwarningswarnrr argmax) r r rrnearest_dists_from_bnearest_a_point_indices_from_bnearest_dists_from_anearest_b_point_indices_from_amax_index_from_amax_index_from_bmax_dist_from_amax_dist_from_brrrhausdorff_pair9s,#  r )rnumpyr scipy.spatialrrr rrrrs   2