khZddlZddlmZddlmZddlmZdgZd dddd Z d Z dd Z d Z y)N)contingency_table)check_shape_equalityvariation_of_informationtable ignore_labelsct||||\}}tj|j|jgS)uReturn symmetric conditional entropies associated with the VI. [1]_ The variation of information is defined as VI(X,Y) = H(X|Y) + H(Y|X). If X is the ground-truth segmentation, then H(X|Y) can be interpreted as the amount of under-segmentation and H(Y|X) as the amount of over-segmentation. In other words, a perfect over-segmentation will have H(X|Y)=0 and a perfect under-segmentation will have H(Y|X)=0. Parameters ---------- image0, image1 : ndarray of int Label images / segmentations, must have same shape. table : scipy.sparse array in csr format, optional A contingency table built with skimage.evaluate.contingency_table. If None, it will be computed with skimage.evaluate.contingency_table. If given, the entropies will be computed from this table and any images will be ignored. ignore_labels : sequence of int, optional Labels to ignore. Any part of the true image labeled with any of these values will not be counted in the score. Returns ------- vi : ndarray of float, shape (2,) The conditional entropies of image1|image0 and image0|image1. References ---------- .. [1] Marina Meilă (2007), Comparing clusterings—an information based distance, Journal of Multivariate Analysis, Volume 98, Issue 5, Pages 873-895, ISSN 0047-259X, :DOI:`10.1016/j.jmva.2006.11.013`. r ) _vi_tablesnparraysum)image0image1r r h0g1h1g0s /home/developers/rajanand/mypropertyqr-fmb-refixing-v2/venv/lib/python3.12/site-packages/skimage/metrics/_variation_of_information.pyrr s9BFF%}UJD$ 88TXXZ, --c|j}tj|r|jdvr |j}nt j |}|j}||xxt j||zcc<|S)aCompute x * log_2(x). We define 0 * log_2(0) = 0 Parameters ---------- x : ndarray or scipy.sparse.csc_array or scipy.sparse.csr_array The input array. Returns ------- y : same type as x Result of x * log_2(x). )csccsr) copysparseissparseformatdatarasarraynonzerolog2)xyznzs r_xlogxr&/sf A qahh.8 FF JJqM BbERWWQrU^E Hrct|||t|||d}n|}tj|j d}tj|j d}t j t|df|j|jf}t j t|df|j|jf}| t||zj dz} t||zj d |z} tttj| | gS)aCompute probability tables used for calculating VI. Parameters ---------- im_true, im_test : ndarray of int Input label images, any dimensionality. table : csr_array, optional Pre-computed contingency table. ignore_labels : sequence of int, optional Labels to ignore when computing scores. Returns ------- hxgy, hygx : ndarray of float Per-segment conditional entropies of ``im_true`` given ``im_test`` and vice-versa. T)r normalizer)axisr)shape) rrrravelrr dia_array_invert_nonzerosizer&listmapr) im_trueim_testr r pxypxpypx_invpy_invhygxhxgys rr r Hs$'* } WMT   #''q'/ "B #''q'/ "B  r2A6rww>P QF   r2A6rww>P QF 3 %))q)1 1D 3< $ $! $ , ,r 1D BJJt - ..rcf|j}tj|}d||z ||<|S)a Compute the inverse of the non-zero elements of arr, not changing 0. Parameters ---------- arr : ndarray Returns ------- arr_inv : ndarray Array containing the inverse of the non-zero elements of arr, and zero elsewhere. r)rrr )arrarr_invr%s rr-r-us2hhjG CBc"g+GBK Nr)NN)Nr) numpyr scipy.sparser_contingency_tabler _shared.utilsr__all__rr&r r-rrrrBs910 % &#.TV#.L 2*/Zr