U !?h8 @s\ddlZddlmZddlZddlZddlmZddlmZ ddl m Z Gddde Z dS)N)List)Image) BaseSessionc@s<eZdZdZeeedddZeddZeddZ d S) DisSessionz? This class represents a session for object detection. )imgreturnc Os|jd||ddd}|ddddddddf}t|}t|}||||}t|}tj|d ddd }| |j tj j }|gS) aI Use a pre-trained model to predict the object in the given image. Parameters: img (PILImage): The input image. *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: List[PILImage]: A list of predicted mask images. N)g ףp= ?gv/?gCl?)?r r )r rZuint8L)mode)Z inner_sessionrun normalizenpmaxminZsqueezerZ fromarrayZastyperesizesizeZ ResamplingZLANCZOS) selfrargskwargsZort_outspredmamimaskrJ/var/www/html/venv/lib/python3.8/site-packages/rembg/sessions/dis_anime.pypredicts     zDisSession.predictcOsT|j||d}tjd|j||r(dnd||j||ddtj|j|||S)z Download the pre-trained models. Parameters: *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: str: The path of the downloaded model file. z.onnxzNhttps://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-anime.onnxNz$md5:6f184e756bb3bd901c8849220a83e38eT)fnamepathZ progressbar)namepoochretrieveZchecksum_disabledZ u2net_homeosr join)clsrrrrrrdownload_models/s    zDisSession.download_modelscOsdS)z Get the name of the pre-trained model. Parameters: *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: str: The name of the pre-trained model. z isnet-animer)r&rrrrrr!Js zDisSession.nameN) __name__ __module__ __qualname____doc__PILImagerr classmethodr'r!rrrrr s  r) r$typingrnumpyrr"ZPILrZ PIL.Imager,baserrrrrrs