# 3jSSKrSSKJrJr \"SSSSS9S5r\"SSSSS9S5r\"S SSS 9S 5rS S jrg)N)njitprangezvoid(f4[:, :, :], f4[:, :, :])T)cachenogilparallelc 8URup#nURupVn[U5Hmn[U5H[n[S[ US- X-U-55n [S[ US- Xr-U-55n [U5H n XX4XX4'M M] Mo g)a Internal method. Resize image src to dst using nearest neighbors filtering. Images must have multiple color channels, i.e. :code:`len(shape) == 3`. Parameters ---------- dst: numpy.ndarray of type np.float32 output image src: numpy.ndarray of type np.float32 input image rNshaperrangemaxmin) dstsrch_srcw_srcdepthh_dstw_dsty_dstx_dstx_srcy_srccs e/home/wildlama/miniconda3/lib/python3.13/site-packages/pymatting/foreground/estimate_foreground_ml.py_resize_nearest_multichannelrs))E%))E%5\E3uqy%-5*@ABE3uqy%-5*@ABE5\'*%?';5O$" "zvoid(f4[:, :], f4[:, :])c URup#URupE[U5H[n[U5HIn[S[ US- Xs-U-55n[S[ US- Xb-U-55n XU4XU4'MK M] g)a  Internal method. Resize image src to dst using nearest neighbors filtering. Images must be grayscale, i.e. :code:`len(shape) == 3`. Parameters ---------- dst: numpy.ndarray of type np.float32 output image src: numpy.ndarray of type np.float32 input image rr Nr ) rrrrrrrrrrs r_resize_nearestr s99LE99LE5\E3uqy%-5*@ABE3uqy%-5*@ABE #5L 1Cu  "rzLTuple((f4[:, :, :], f4[:, :, :]))(f4[:, :, :], f4[:, :], f4, i4, i4, i4, f4))rrctURupxn [Rn Sn Sn [R"XS9n [R"XS9nSnSn[ U5Hn[ U5HonUUU4S:a*[ U 5HnU U==UUUU4- ss'M US- nUUU4S:dME[ U 5HnUU==UUUU4- ss'M US- nMq M XS--n UUS--n[R"XU 4U S9U -n[R"XU 4U S9U-n[ [R "[R"[X5555n[ US-5GHn[UUU- -5n[UUU- -5n[R"UUU 4U S9n[R"UU4U S9n[UU5 [UU5 [R"UUU 4U S9n[R"UUU 4U S9n[UU5 [UU5 UU::a UU::aUnOUn[R"SU 4U S9n/SQn /S Qn![ U5GHn"[U5GHn[ U5GHnUUU4n#S U#- n$U#U#-n%U#U$-n&U$U$-n'[ U 5H#nU#UUUU4-USU4'U$UUUU4-USU4'M% [ S 5Hn([S[US- UU U(-55n)[S[US- UU!U(-55n*[U#UU*U)4- 5n+X&U+--n,U%U,- n%U'U,- n'[ U 5H3nUSU4==U,UU*U)U4-- ss'USU4==U,UU*U)U4-- ss'M5 M U%U'-U&U&-- n-S U-- n.U.U'-n/U.U&*-n0U.U%-n1[ U 5HinU/USU4-U0USU4--n2U0USU4-U1USU4--n3[S [S U255n2[S [S U355n3U2UUUU4'U3UUUU4'Mk GM GM GM UnUnUn Un GM WW4$) Nr )dtyperg?g?h㈵>)r rr)rrr$r ?g)r npfloat32zerosr intceillog2r roundemptyrrrrabs)4 input_image input_alpharegularizationn_small_iterationsn_big_iterations small_sizegradient_weighth0w0rr!w_prevh_prev F_mean_color B_mean_colorF_countB_countyxrF_prevB_prevn_levelsi_levelwhimagealphaFBn_iterbdxdyi_itera0a1a00a01a11dx2y2gradientda determinantinv_detb00b01b11F_cB_cs4 r_estimate_fb_mlra=s %%MBE JJE F F88E/L88E/LGG 2YrA1a4 3&uA O{1a7';;O&1 1a4 3&uA O{1a7';;O&1 dN"LGdN"LXXvu-U ;l JF XXvu-U ;l JF2772773r;/01HA& "8+, - "8+, -!Qe4!Qu-$UK8{+ HHaE]% 0 HHaE]% 0$Q/$Q/ ?qJ'F%F HHaZu -  FmFAYqAq!tBrBr'Cr'Cr'C"5\"$uQ1W~"5!Q$"$uQ1W~"5!Q$*#1X CAq2a5y$9: CAq2a5y$9:#&rE"b&M'9#:+.HHr r !&uAadGrAb"aiL'88GadGrAb"aiL'88G".&#&)cCi"7K!K/G!C-C!SD.C!C-C"5\!AadGmcAadGm;!AadGmcAadGm;!#s3}5!#s3}5%(!Q' %(!Q' *I"$`]'` a4Krc [UR[R5UR[R5UUUUU5upU(aX4$U$)aEstimates the foreground of an image given its alpha matte. See :cite:`germer2020multilevel` for reference. Parameters ---------- image: numpy.ndarray Input image with shape :math:`h \times w \times d` alpha: numpy.ndarray Input alpha matte shape :math:`h \times w` regularization: float Regularization strength :math:`\epsilon`, defaults to :math:`10^{-5}`. Higher regularization results in smoother colors. n_small_iterations: int Number of iterations performed on small scale, defaults to :math:`10` n_big_iterations: int Number of iterations performed on large scale, defaults to :math:`2` small_size: int Threshold that determines at which size `n_small_iterations` should be used return_background: bool Whether to return the estimated background in addition to the foreground gradient_weight: float Larger values enforce smoother foregrounds, defaults to :math:`1` Returns ------- F: numpy.ndarray Extracted foreground B: numpy.ndarray Extracted background Example ------- >>> from pymatting import * >>> image = load_image("data/lemur/lemur.png", "RGB") >>> alpha = load_image("data/lemur/lemur_alpha.png", "GRAY") >>> F = estimate_foreground_ml(image, alpha, return_background=False) >>> F, B = estimate_foreground_ml(image, alpha, return_background=True) See Also -------- stack_images: This function can be used to place the foreground on a new background. )raastyper'r() rGrHr2r3r4r5return_backgroundr6 foreground backgrounds restimate_foreground_mlrgsUl- RZZ  RZZ J%% r)r" r# Fr%) numpyr'numbarrrrrargrrrms&d$N<O<4 D4H2I28T\`hlmyny~Cr