# 3j bSSKrSSKJr SSKrSSKrSSKJr SSKJr SSK J r "SS\ 5r g)N)List)Image) BaseSessioncL\rSrSrS\S\\4Sjr\S5r\S5r Sr g) DisSession imgreturnc URRSURUSSS55nUSSS2SSS2SS24n[R"U5n[R "U5nXW- Xg- - n[R "U5n[R"US-RS5SS 9nURUR[RR5nU/$) al Predicts the mask image for the input image. This method takes a PILImage object as input and returns a list of PILImage objects as output. It performs several image processing operations to generate the mask image. Parameters: img (PILImage): The input image. Returns: List[PILImage]: A list of PILImage objects representing the generated mask image. N)?r r )?rr)rruint8L)mode) inner_sessionrun normalizenpmaxminsqueezer fromarrayastyperesizesize ResamplingLANCZOS) selfr argskwargsort_outspredmamimasks X/home/wildlama/miniconda3/lib/python3.13/site-packages/rembg/sessions/dis_general_use.pypredictDisSession.predict s%%))  NN3, O  {1aA:& VVD\ VVD\ bg&zz$s 227;#F{{388U%5%5%=%=>v c UR"U0UD6S3n[R"SUR"U0UD6(aSOSUUR"U0UD6SS9 [ R RUR"U0UD6U5$)a) Downloads the pre-trained model file. This class method downloads the pre-trained model file from a specified URL using the pooch library. Parameters: args: Additional positional arguments. kwargs: Additional keyword arguments. Returns: str: The path to the downloaded model file. z.onnxzThttps://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-general-use.onnxNz$md5:fc16ebd8b0c10d971d3513d564d01e29T)fnamepath progressbar)namepoochretrievechecksum_disabled u2net_homeosr/join)clsr"r#r.s r)download_modelsDisSession.download_models+s88T,V,-U3  b(($9&9;00 ww||CNND;F;UCCr,cg)z Returns the name of the model. This class method returns the name of the model. Parameters: args: Additional positional arguments. kwargs: Additional keyword arguments. Returns: str: The name of the model. zisnet-general-use)r8r"r#s r)r1DisSession.nameHs#r,r<N) __name__ __module__ __qualname____firstlineno__PILImagerr* classmethodr9r1__static_attributes__r<r,r)rr sC8h<DD8 # #r,r) r6typingrnumpyrr2PILr PIL.ImagerBbaserrr<r,r)rJs(  'J#J#r,