# 3ju bSSKrSSKJr SSKrSSKrSSKJr SSKJr SSK J r "SS\ 5r g)N)List)Image) BaseSessioncP\rSrSrSrS\S\\4Sjr\S5r \S5r Sr g ) U2netHumanSegSession zZ This class represents a session for performing human segmentation using the U2Net model. imgreturnc URRSURUSSS55nUSSS2SSS2SS24n[R"U5n[R "U5nXW- Xg- - n[R "U5n[R"US-RS5SS 9nURUR[RR5nU/$) z Predicts human segmentation masks for the input image. Parameters: img (PILImage): The input image. *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: List[PILImage]: A list of predicted masks. N)g ףp= ?gv/?gCl?)gZd;O?gy&1?g?)@r ruint8L)mode) inner_sessionrun normalizenpmaxminsqueezer fromarrayastyperesizesize ResamplingLANCZOS) selfr argskwargsort_outspredmamimasks X/home/wildlama/miniconda3/lib/python3.13/site-packages/rembg/sessions/u2net_human_seg.pypredictU2netHumanSegSession.predicts%%))  NN*,A:  {1aA:& VVD\ VVD\ bg&zz$s 227;#F{{388U%5%5%=%=>v c UR"U0UD6S3n[R"SUR"U0UD6(aSOSUUR"U0UD6SS9 [ R RUR"U0UD6U5$)z Downloads the U2Net model weights. Parameters: *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: str: The path to the downloaded model weights. z.onnxzRhttps://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnxNz$md5:c09ddc2e0104f800e3e1bb4652583d1fT)fnamepath progressbar)namepoochretrievechecksum_disabled u2net_homeosr-join)clsr r!r,s r'download_models$U2netHumanSegSession.download_models1s88T,V,-U3  `(($9&9;00 ww||CNND;F;UCCr*cg)z Returns the name of the U2Net model. Parameters: *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: str: The name of the model. u2net_human_seg)r6r r!s r'r/U2netHumanSegSession.nameLs!r*r;N) __name__ __module__ __qualname____firstlineno____doc__PILImagerr( classmethodr7r/__static_attributes__r;r*r'rr sI8h@DD4 ! !r*r) r4typingrnumpyrr0PILr PIL.ImagerBbaserrr;r*r'rJs(  'L!;L!r*