1-Ph5ddlZddlmZddlmZdZddZddZ d Z d Z d Z d Z d ZejfdZdZdZejfdZ ddZdS)N)distance_transform_edt)_supported_float_typec^d}tj|dd}|ddddf|ddddfz }|ddddf|ddddfz } |ddddf|ddddfz d z } |ddddf|ddddfz } |ddddf|ddddfz } |ddddf|ddddfz d z } d tj||dzz| dzzz }d tj|| dzz| dzzz }d tj|| dzz| dzzz }d tj|| dzz| dzzz }|ddddf|z|ddddf|zz|ddddf|zz|ddddf|zz}|d k|j}t ||\}}| ||z dzz|||z dzzz}||t |z||z|zzz}|d||zt |z||z|z|zzzz S) aReturns the variation of level set 'phi' based on algorithm parameters. This corresponds to equation (22) of the paper by Pascal Getreuer, which computes the next iteration of the level set based on a current level set. A full explanation regarding all the terms is beyond the scope of the present description, but there is one difference of particular import. In the original algorithm, convergence is accelerated, and required memory is reduced, by using a single array. This array, therefore, is a combination of non-updated and updated values. If this were to be implemented in python, this would require a double loop, where the benefits of having fewer iterations would be outweided by massively increasing the time required to perform each individual iteration. A similar approach is used by Rami Cohen, and it is from there that the C1-4 notation is taken. gؗҜ<edgemoderN@?r)nppadsqrtastypedtype_cv_calculate_averages _cv_delta)imagephimulambda1lambda2dtetaPphixpphixnphix0phiypphiynphiy0C1C2C3C4KHphic1c2difference_from_average_termnew_phis _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/segmentation/_chan_vese.py_cv_calculate_variationr/s$ C sAF###A adABBhK!AbD!B$J- 'E adAbDjMAadCRCiL (E qtQRRx[1QrT3B3Y< '3 .E abb!B$hK!AbD!B$J- 'E adAbDjMAcrc1R4iL (E qrr1R4x[1SbS!B$Y< '3 .E rwsUAX~q011 1B rwsUAX~q011 1B rwsUAX~q011 1B rwsUAX~q011 1B !B$( b1QrT3B3Y<",,qQrT{R/??!CRC2I,QSBSSA !G  EK ( (D%eT22HR EBJ1$$w%"*1B'BB!R)C..(R!V6R-RSSG a"r'IcNN2b2glR6GHH IIrc\dddtjz tj||z zzzS)zVReturns the result of a regularised heavyside function of the input value(s). ?rr )rpiarctanxepss r. _cv_heavysider85s- #ru !c'(:(::: ;;r0c||dz|dzzz S)zRReturns the result of a regularised dirac function of the input value(s). rr5s r.rr<s #q&1a4-  r0c|}d|z }tj|}tj|}tj||z}tj||z}|dkr||z}|dkr||z}||fS)z2Returns the average values 'inside' and 'outside'.rr)rsum)rr)HHinvHsumHinvsum avg_inside avg_oustides r.rrCs A 7D 6!99DfTllG ""J&&&K qyyd !||w  $$r0clt||\}}d|z }|||z dzz|z|||z dzz|zzS)zvReturns the 'energy' contribution due to the difference from the average value within a region at each point. rr)r)rr) lambda_pos lambda_negr*r+r>s r. _cv_difference_from_average_termrFRsS&eT22HR :D ) )D 0:PQ@Q3QTX3X XXr0ctj|dd}|ddddf|ddddfz dz }|ddddf|ddddfz dz }|t|ztj|dz|dzzzS) zReturns the 'energy' contribution due to the length of the edge between regions at each point, multiplied by a factor 'mu'. rrr rNr r r )rrrr)rrrfyfxs r._cv_edge_length_termrJ[s sAF###A ABB"H+#2#qt) $ +B AbD!""H+!B$) $ +B # QQ!7!7 77r0ct|}t||||}t||}tj|tj|zS)aWReturns the total 'energy' of the current level set function. This corresponds to equation (7) of the paper by Pascal Getreuer, which is the weighted sum of the following: (A) the length of the contour produced by the zero values of the level set, (B) the area of the "foreground" (area of the image where the level set is positive), (C) the variance of the image inside the foreground, (D) the variance of the image outside of the foreground Each value is computed for each pixel, and then summed. The weight of (B) is set to 0 in this implementation. )r8rFrJrr<)rrrrrr= avgenergy lenenergys r. _cv_energyrNesR cA07GLLI$S"--I 6)  rvi00 00r0c|S)zThis is a placeholder function as resetting the level set is not strictly necessary, and has not been done for this implementation. r:)rs r._cv_reset_level_setrPzs  Jr0c*tj|d||dd}tj|d|}tj|z }||z}||z}tj|tj|zS)zGenerates a checkerboard level set function. According to Pascal Getreuer, such a level set function has fast convergence. rrr)rarangereshaper3sin) image_size square_sizeryvxvsfs r._cv_checkerboardr[s :a= . . . 6 6z!}a H HB :a= . . .B  B"HB"HB 6"::r ""r0ctj|}t|ddz dz }t|ddz dz }d|||f<tt ||}|t |z |z S)zlGenerates a disk level set function. The disk covers the whole image along its smallest dimension. rrrronesintfloatmindistancerVrescenterYcenterXradiuss r._cv_large_diskris '*  C:a=1$)**G:a=1$)**GC 3w(( ) )F Xc]] "f ,,r0ctj|}t|ddz dz }t|ddz dz }d|||f<tt ||dz }|t |z |dzz S)znGenerates a disk level set function. The disk covers half of the image along its smallest dimension. rrrr]r r^rds r._cv_small_diskrls '*  C:a=1$)**G:a=1$)**GC 3w(( ) )C /F Xc]] "vz 22r0ct|trS|dkrt|d|}n=|dkrt|}n'|dkrt |}nt d|}||dS)zGGenerates an initial level set function conditional on input arguments. checkerboarddiskz small diskz-Incorrect name for starting level set preset.Fcopy) isinstancestrr[rirl ValueErrorr)init_level_set image_shaperres r._cv_init_level_setrxs.#&&  ^ + +";599CC v % % --CC | + + --CCLMM M ::e%: ( ((r0?MbP?r2rnFc jt|jdkrtdt|j} t ||j| } t | tjks| j|jkrtd| | d}|tj |z }tj |dkr|tj |z }d} t|| |||} g} |dz}| dk}||kr| |kr| }t|| ||||} t| } tj| |z dz}| dk}t|| |||}| | |} | dz } ||kr| |k|r|| | fS|S) aChan-Vese segmentation algorithm. Active contour model by evolving a level set. Can be used to segment objects without clearly defined boundaries. Parameters ---------- image : (M, N) ndarray Grayscale image to be segmented. mu : float, optional 'edge length' weight parameter. Higher `mu` values will produce a 'round' edge, while values closer to zero will detect smaller objects. lambda1 : float, optional 'difference from average' weight parameter for the output region with value 'True'. If it is lower than `lambda2`, this region will have a larger range of values than the other. lambda2 : float, optional 'difference from average' weight parameter for the output region with value 'False'. If it is lower than `lambda1`, this region will have a larger range of values than the other. tol : float, positive, optional Level set variation tolerance between iterations. If the L2 norm difference between the level sets of successive iterations normalized by the area of the image is below this value, the algorithm will assume that the solution was reached. max_num_iter : uint, optional Maximum number of iterations allowed before the algorithm interrupts itself. dt : float, optional A multiplication factor applied at calculations for each step, serves to accelerate the algorithm. While higher values may speed up the algorithm, they may also lead to convergence problems. init_level_set : str or (M, N) ndarray, optional Defines the starting level set used by the algorithm. If a string is inputted, a level set that matches the image size will automatically be generated. Alternatively, it is possible to define a custom level set, which should be an array of float values, with the same shape as 'image'. Accepted string values are as follows. 'checkerboard' the starting level set is defined as sin(x/5*pi)*sin(y/5*pi), where x and y are pixel coordinates. This level set has fast convergence, but may fail to detect implicit edges. 'disk' the starting level set is defined as the opposite of the distance from the center of the image minus half of the minimum value between image width and image height. This is somewhat slower, but is more likely to properly detect implicit edges. 'small disk' the starting level set is defined as the opposite of the distance from the center of the image minus a quarter of the minimum value between image width and image height. extended_output : bool, optional If set to True, the return value will be a tuple containing the three return values (see below). If set to False which is the default value, only the 'segmentation' array will be returned. Returns ------- segmentation : (M, N) ndarray, bool Segmentation produced by the algorithm. phi : (M, N) ndarray of floats Final level set computed by the algorithm. energies : list of floats Shows the evolution of the 'energy' for each step of the algorithm. This should allow to check whether the algorithm converged. Notes ----- The Chan-Vese Algorithm is designed to segment objects without clearly defined boundaries. This algorithm is based on level sets that are evolved iteratively to minimize an energy, which is defined by weighted values corresponding to the sum of differences intensity from the average value outside the segmented region, the sum of differences from the average value inside the segmented region, and a term which is dependent on the length of the boundary of the segmented region. This algorithm was first proposed by Tony Chan and Luminita Vese, in a publication entitled "An Active Contour Model Without Edges" [1]_. This implementation of the algorithm is somewhat simplified in the sense that the area factor 'nu' described in the original paper is not implemented, and is only suitable for grayscale images. Typical values for `lambda1` and `lambda2` are 1. If the 'background' is very different from the segmented object in terms of distribution (for example, a uniform black image with figures of varying intensity), then these values should be different from each other. Typical values for mu are between 0 and 1, though higher values can be used when dealing with shapes with very ill-defined contours. The 'energy' which this algorithm tries to minimize is defined as the sum of the differences from the average within the region squared and weighed by the 'lambda' factors to which is added the length of the contour multiplied by the 'mu' factor. Supports 2D grayscale images only, and does not implement the area term described in the original article. References ---------- .. [1] An Active Contour Model without Edges, Tony Chan and Luminita Vese, Scale-Space Theories in Computer Vision, 1999, :DOI:`10.1007/3-540-48236-9_13` .. [2] Chan-Vese Segmentation, Pascal Getreuer Image Processing On Line, 2 (2012), pp. 214-224, :DOI:`10.5201/ipol.2012.g-cv` .. [3] The Chan-Vese Algorithm - Project Report, Rami Cohen, 2011 :arXiv:`1107.2782` rz!Input image should be a 2D array.rRzIThe dimensions of initial level set do not match the dimensions of image.Frqrr)lenshaperurrrxtyperndarrayrrbmaxrNr/rPrmeanappend)rrrrtol max_num_iterrrvextended_output float_dtyperi old_energyenergiesphivar segmentationoldphi new_energys r. chan_vesersN 5;1<===' 44K ^U[ L L LC CyyBJ#)u{":": -   LL5L 1 1E BF5MM !E ve}}u % AE3GW==JH 1WF7L 3,,1|++&eS"gwKK!#&&3rsP<<<<<<111111+J+J+J\<<<<!!!! % % %YYY888111*57J # # # # - - - 3 3 3;=* ) ) ) )$    !ttttttr0