o
    ;ήc                     @   sL   d dl mZ d dlZd dlmZ d dlmZ ddlm	Z	 G dd de	Z
dS )    )ListN)Image   )BaseSessionc                   @   s"   e Zd Zdedee fddZdS )SimpleSessionimgreturnc              	   C   s   | j d | |ddd}|d d d dd d d d f }t|}t|}|| ||  }t|}tj|d 	ddd}|
|jtj}|gS )	N)g
ףp=
?gv/?gCl?)gZd;O?gy&1?g?)@  r	   r      uint8L)mode)inner_sessionrun	normalizenpmaxminsqueezer   	fromarrayastyperesizesizeLANCZOS)selfr   ort_outspredmamimask r    ;/tmp/pip-target-vg8gfxp4/lib/python/rembg/session_simple.pypredict   s    


zSimpleSession.predictN)__name__
__module____qualname__PILImager   r"   r    r    r    r!   r   
   s    r   )typingr   numpyr   PILr   	PIL.Imager&   session_baser   r   r    r    r    r!   <module>   s    