U !?h @s\ddlZddlmZddlZddlZddlmZddlmZ ddl m Z Gddde Z dS)N)List)Image) BaseSessionc@s<eZdZdZeeedddZeddZeddZ d S) SiluetaSessionz5This is a class representing a SiluetaSession object.)imgreturnc Os|jd||ddd}|ddddddddf}t|}t|}||||}t|}tj|d ddd }| |j tj j }|gS) a Predict the mask of the input image. This method takes an image as input, preprocesses it, and performs a prediction to generate a mask. The generated mask is then post-processed and returned as a list of PILImage objects. Parameters: img (PILImage): The input image to be processed. *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: List[PILImage]: A list of post-processed masks. N)g ףp= ?gv/?gCl?)gZd;O?gy&1?g?)@r rZuint8L)mode)Z inner_sessionrun normalizenpmaxminZsqueezerZ fromarrayZastyperesizesizeZ ResamplingZLANCZOS) selfrargskwargsZort_outspredmamimaskrH/var/www/html/venv/lib/python3.8/site-packages/rembg/sessions/silueta.pypredicts"    zSiluetaSession.predictcOsP|d}tjd|j||r$dnd||j||ddtj|j|||S)ay Download the pre-trained model file. This method downloads the pre-trained model file from a specified URL. The file is saved to the U2NET home directory. Parameters: *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: str: The path to the downloaded model file. z.onnxzJhttps://github.com/danielgatis/rembg/releases/download/v0.0.0/silueta.onnxNz$md5:55e59e0d8062d2f5d013f4725ee84782T)fnamepathZ progressbar)namepoochretrieveZchecksum_disabledZ u2net_homeosrjoin)clsrrrrrrdownload_models1s   zSiluetaSession.download_modelscOsdS)a Return the name of the model. This method returns the name of the Silueta model. Parameters: *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: str: The name of the model. Zsiluetar)r%rrrrrr NszSiluetaSession.nameN) __name__ __module__ __qualname____doc__PILImagerr classmethodr&r rrrrr s " r) r#typingrnumpyrr!ZPILrZ PIL.Imager+baserrrrrrs