1-Ph\ddlZddlmZddlmZddlmZddlm Z  dd ddZ dS)N)RectBivariateSpline)_supported_float_type) img_as_float)sobel{Gz?皙?? periodic)boundary_conditionc   t|}|dkrtdd} gd} | | vr(tdd| zdzt|} t |j}| |d } | jd k}|dkrg|rTt| d d d d dft| d d d d d ft| d d d d d fg}nt| g}ndg}|r-|tj | d z|t|zz} n|| z||dzz} ttj | j d tj | j d| jd d d}|d d d d df}|d d df|}|d d d f|}t|}tj| |f|}tj| |f|}tj|t$}tj|ddtj|dd zd |zz }tj|ddtj|dd zdtj|ddzz dtj|dd zz d|zz}| |z||zz}d}| drd|dd d f<d|d d d f<gd|d d d f<d}d}| drd|dd d f<d|dd d f<gd|ddd f<d}d}| dr*d|dd d f<gd|dd d f<d|d d d f<gd|d d df<d}d}| dr*d|dd d f<gd|ddd f<d|dd d f<gd|ddd f<d}tj|||zz} | |d } t1|D]}!|||d d|d }"|||d d|d }#|r d|"d<d|#d<|r d|"d<d|#d<|r |"dxxd zcc<|#dxxd zcc<|r |"dxxd zcc<|#dxxd zcc<| ||z|"zz}$| ||z|#zz}%|tj|$|z z}&|tj|%|z z}'|r d|&d<d|'d<|r d|&d<d|'d<||&z }||'z }|!| d zz}(|(| kr|||(d d f<|||(d d f<5tjtjtj||d d d fz tj||d d d fz zd })|)| krntj||gd S)a Active contour model. Active contours by fitting snakes to features of images. Supports single and multichannel 2D images. Snakes can be periodic (for segmentation) or have fixed and/or free ends. The output snake has the same length as the input boundary. As the number of points is constant, make sure that the initial snake has enough points to capture the details of the final contour. Parameters ---------- image : (M, N) or (M, N, 3) ndarray Input image. snake : (K, 2) ndarray Initial snake coordinates. For periodic boundary conditions, endpoints must not be duplicated. alpha : float, optional Snake length shape parameter. Higher values makes snake contract faster. beta : float, optional Snake smoothness shape parameter. Higher values makes snake smoother. w_line : float, optional Controls attraction to brightness. Use negative values to attract toward dark regions. w_edge : float, optional Controls attraction to edges. Use negative values to repel snake from edges. gamma : float, optional Explicit time stepping parameter. max_px_move : float, optional Maximum pixel distance to move per iteration. max_num_iter : int, optional Maximum iterations to optimize snake shape. convergence : float, optional Convergence criteria. boundary_condition : string, optional Boundary conditions for the contour. Can be one of 'periodic', 'free', 'fixed', 'free-fixed', or 'fixed-free'. 'periodic' attaches the two ends of the snake, 'fixed' holds the end-points in place, and 'free' allows free movement of the ends. 'fixed' and 'free' can be combined by parsing 'fixed-free', 'free-fixed'. Parsing 'fixed-fixed' or 'free-free' yields same behaviour as 'fixed' and 'free', respectively. Returns ------- snake : (K, 2) ndarray Optimised snake, same shape as input parameter. References ---------- .. [1] Kass, M.; Witkin, A.; Terzopoulos, D. "Snakes: Active contour models". International Journal of Computer Vision 1 (4): 321 (1988). :DOI:`10.1007/BF00133570` Examples -------- >>> from skimage.draw import circle_perimeter >>> from skimage.filters import gaussian Create and smooth image: >>> img = np.zeros((100, 100)) >>> rr, cc = circle_perimeter(35, 45, 25) >>> img[rr, cc] = 1 >>> img = gaussian(img, sigma=2, preserve_range=False) Initialize spline: >>> s = np.linspace(0, 2*np.pi, 100) >>> init = 50 * np.array([np.sin(s), np.cos(s)]).T + 50 Fit spline to image: >>> snake = active_contour(img, init, w_edge=0, w_line=1) # doctest: +SKIP >>> dist = np.sqrt((45-snake[:, 0])**2 + (35-snake[:, 1])**2) # doctest: +SKIP >>> int(np.mean(dist)) # doctest: +SKIP 25 rzmax_num_iter should be >0. )rfreefixedz free-fixedz fixed-freez fixed-fixedz free-freez.Invalid boundary condition. Should be one of: z, .F)copyNr r)axis)kxkys)dtyper)r rr Tr)rrr r )dxgrid)dyr#)int ValueErrorjoinrrrastypendimrnpsumrarangeshapeTlenemptyeyefloatroll startswithendswithlinalginvrangetanhminmaxabsstack)*imagesnakealphabetaw_linew_edgegamma max_px_move max_num_iter convergencerconvergence_order valid_bcsimg float_dtypeRGBedgeintpsnake_xyxynxsaveysaveeye_nabAsfixedefixedsfreeefreer7ifxfyxnynr"r$jdists* i/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/segmentation/active_contour_model.pyactive_contourre s}||$$Lq5666I** #ii "" #     u  C' 44K **[u* - -C (a-C{{  #aaaAg,''s111aaa7|)<)Q>QRDD#JJ>!RaR% E""6**"aaa%ZZ"bcc' "aaa%#^^"bcc'  )--EEM) * *C **[u* - -C< ++ T!Q15 ) ) ) 0 05 0 I I T!Q15 ) ) ) 0 05 0 I I  BqEBqE  BrFBrF   qEEEQJEEE qEEEQJEEE   rFFFaKFFF rFFFaKFFF EAIN # EAIN #2726?? * 2726?? *  BqEBqE  BrFBrF R R "Q& '  E!QQQ$KE!QQQ$KK6rveaaaaj011BF51T111W:;M4N4NNPQRRDk!!" 8QF # # ##)rr r r rr r r ) numpyr*scipy.interpolater _shared.utilsrutilrfiltersrrerfrdrms111111111111   q$"q$q$q$q$q$q$q$rf