
    
3j                     B    S SK rS SKJr  SSKJrJr   " S S5      r\/rg)    N)Image   )
pil2tensor
tensor2pilc                   >    \ rS rSrSr\S 5       rSrSrSr	Sr
S rS	rg
)MTB_ImageRemoveBackgroundRembg   z2Removes the background from the input using Rembg.c                 X    SSSSS04SSSS	S
.4SSSS	S
.4SSSS	S
.4SSS04SSSS.4S.0$ )Nrequired)IMAGEBOOLEANdefaultFINT   r      )r   minmax
   COLORz#000000	MTB_COLOR)r   
widgetType)imagealpha_matting"alpha_matting_foreground_threshold"alpha_matting_background_thresholdalpha_matting_erode_sizepost_process_maskbgcolor )clss    A/home/wildlama/comfy/ComfyUI/custom_nodes/comfy_mtb/nodes/mask.pyINPUT_TYPES*MTB_ImageRemoveBackgroundRembg.INPUT_TYPES
   s     #&"
  #Ac:7
  "1S97
  "1S9-
 &&
  )D-
 	
    )r   MASKr   )zImage (rgba)Maskr   remove_backgroundz	mtb/imagec                 *   SSK Jn  [        R                  R	                  UR                  S5      5      n	[        U5      n
/ n/ n/ nU
 H  nU" UUUUUS SUS S9	nUR                  S5      n[        R                  " SUR
                  U5      nUR                  UUS9  UR                  S5      nUR                  U5        UR                  U5        UR                  U5        U	R                  S	5        M     [        U5      [        U5      [        U5      4$ )
Nr   )removeF)	datar   r   r   r   session	only_maskr   r      RGBA)maskRGB   )rembgr)   comfyutilsProgressBarsizer   
getchannelr   newpasteconvertappendupdater   )selfr   r   r   r   r   r   r   r)   pbarimagesout_imgout_maskout_img_on_bgimgimg_rmr/   image_on_bgs                     r!   r'   0MTB_ImageRemoveBackgroundRembg.remove_background8   s    	!{{&&uzz!}5E"C+3U3U)A"3
F $$Q'D  ))FFKKAKf40%--e4KNN6"OOD!  -KKN7 < wx }%
 	
r$   r   N)__name__
__module____qualname____firstlineno____doc__classmethodr"   RETURN_TYPESRETURN_NAMESFUNCTIONCATEGORYr'   __static_attributes__r   r$   r!   r   r      s6    <
 
>L
L
 #HH4
r$   r   )	comfy.utilsr3   PILr   r4   r   r   r   	__nodes__r   r$   r!   <module>rU      s'      *e
 e
R #	r$   