# 3j-SSKrSSKrSSKJr SSKJrJrJrJrJ r J r SSK r SSK rSS KJrJr SS KJr SS KJr SS KJr SSKJr SSKJrJr SSK J!r!J"r" S SK#J$r$ S SK%J%r%J&r& S SK'J(r( \RR"S5 \!"S 5r*"SS\5r+S\S\S\,S\,S\,S\4 Sjr-S\S\S\4Sjr.S\S\S\4Sjr/S \\S\4S!jr0S"\S#\S\4S$jr1S\ RdS\ Rd4S%jr3S\S&\\,\,\,\,4S\4S'jr4S\S\4S(jr5S)\6\7S*4SS4S+jr8S9S,\ \9\\ Rd4S-\:S.\,S/\,S0\,S1\\(S2\:S3\:S4\\\,\,\,\,4S5\:S6\\S7\\S\ \9\\ Rd44S8jjr;g!\aL \"S5 \"S5 \"5 \"S5 \"S5 \"5 \"S5 \R""S 5 GNf=f):N)Enum)AnyListOptionalTupleUnioncastzNo onnxruntime backend found.z-Please install rembg with CPU or GPU support:z' pip install "rembg[cpu]" # for CPUz3 pip install "rembg[gpu]" # for NVIDIA/CUDA GPUzLFor more information, see: https://github.com/danielgatis/rembg#installation)ImageImageOps)r )estimate_alpha_cf)estimate_foreground_ml) stack_images)binary_erosiongaussian_filter)diskopening) new_session)sessionssessions_names) BaseSessionc \rSrSrSrSrSrSrg) ReturnType'rr N)__name__ __module__ __qualname____firstlineno__BYTESPILLOWNDARRAY__static_attributes__rB/home/wildlama/miniconda3/lib/python3.13/site-packages/rembg/bg.pyrr's E FGr&rimgmaskforeground_thresholdbackground_thresholderode_structure_sizereturncURS:XdURS:XaURS5n[R"U5n[R"U5nXb:nXc:nSn US:a$[R"XD4[R S9n [ XyS9n[ XSS 9n[R"UR[R S S 9n S X'SX'US - n U S - n [X5n [X5n[X5n[R"US -SS 5R[R 5n[R"U5nU$)a Perform alpha matting on an image using a given mask and threshold values. This function takes a PIL image `img` and a PIL image `mask` as input, along with the `foreground_threshold` and `background_threshold` values used to determine foreground and background pixels. The `erode_structure_size` parameter specifies the size of the erosion structure to be applied to the mask. The function returns a PIL image representing the cutout of the foreground object from the original image. RGBACMYKRGBNr)dtype) structurer )r3 border_value)r2 fill_valuego@)modeconvertnpasarrayonesuint8rfullshaper rrclipastyper fromarray)r(r)r*r+r, img_array mask_array is_foreground is_backgroundr3trimapimg_normalizedtrimap_normalizedalpha foregroundcutouts r'alpha_matting_cutoutrM-s*$ xx6SXX/kk%  3ID!J5M5MIaGG ! 8 #=FM"=TUVM WWZ%%RXX# FFFF&N n @E'>J * ,F WWVc\1c * 1 1"(( ;F __V $F Mr&cx[R"SURS5n[R"XU5nU$)aG Perform a simple cutout operation on an image using a mask. This function takes a PIL image `img` and a PIL image `mask` as input. It uses the mask to create a new image where the pixels from `img` are cut out based on the mask. The function returns a PIL image representing the cutout of the original image using the mask. r/r)r newsize composite)r(r)emptyrLs r' naive_cutoutrSbs/ IIfsxx! ,E __S .F Mr&c(URU5 U$)z Apply the specified mask to the image as an alpha cutout. Args: img (PILImage): The image to be modified. mask (PILImage): The mask to be applied. Returns: PILImage: The modified image with the alpha cutout applied. )putalpha)r(r)s r'putalpha_cutoutrVrsLL Jr&imgscPURS5nUHn[X5nM U$)z Concatenate multiple images vertically. Args: imgs (List[PILImage]): The list of images to be concatenated. Returns: PILImage: The concatenated image. r)pop get_concat_v)rWpivotims r'get_concat_v_multir]s* HHQKEU' Lr&img1img2c[R"SURURUR-45nUR US5 UR USUR45 U$)z Concatenate two images vertically. Args: img1 (PILImage): The first image. img2 (PILImage): The second image to be concatenated below the first image. Returns: PILImage: The concatenated image. r/)rrr)r rOwidthheightpaste)r^r_dsts r'rZrZsT ))FTZZt{{)BC DCIIdFIIdQ $% Jr&c[U[5n[UR[R 5SS9n[R "US:SS5R[R5nU$)z Post Process the mask for a smooth boundary by applying Morphological Operations Research based on paper: https://www.sciencedirect.com/science/article/pii/S2352914821000757 args: mask: Binary Numpy Mask r)sigmarr7)rkernelrrAr:float64wherer=)r)s r' post_processrksS 4 D 4;;rzz2! "U5$[R"5nURAUS5 URCS5 URE5$![a$ U(a[3U U5nN[5U U5nNf=f)aW Remove the background from an input image. This function takes in various parameters and returns a modified version of the input image with the background removed. The function can handle input data in the form of bytes, a PIL image, or a numpy array. The function first checks the type of the input data and converts it to a PIL image if necessary. It then fixes the orientation of the image and proceeds to perform background removal using the 'u2net' model. The result is a list of binary masks representing the foreground objects in the image. These masks are post-processed and combined to create a final cutout image. If a background color is provided, it is applied to the cutout image. The function returns the resulting cutout image in the format specified by the input 'return_type' parameter or as python bytes if force_return_bytes is true. Parameters: data (Union[bytes, PILImage, np.ndarray]): The input image data. alpha_matting (bool, optional): Flag indicating whether to use alpha matting. Defaults to False. alpha_matting_foreground_threshold (int, optional): Foreground threshold for alpha matting. Defaults to 240. alpha_matting_background_threshold (int, optional): Background threshold for alpha matting. Defaults to 10. alpha_matting_erode_size (int, optional): Erosion size for alpha matting. Defaults to 10. session (Optional[BaseSession], optional): A session object for the 'u2net' model. Defaults to None. only_mask (bool, optional): Flag indicating whether to return only the binary masks. Defaults to False. post_process_mask (bool, optional): Flag indicating whether to post-process the masks. Defaults to False. bgcolor (Optional[Tuple[int, int, int, int]], optional): Background color for the cutout image. Defaults to None. force_return_bytes (bool, optional): Flag indicating whether to return the cutout image as bytes. Defaults to False. *args (Optional[Any]): Additional positional arguments. **kwargs (Optional[Any]): Additional keyword arguments. Returns: Union[bytes, PILImage, np.ndarray]: The cutout image with the background removed. z^Input type {} is not supported. Try using force_return_bytes=True to force python bytes outputrUFu2netrPNG)# isinstancebytesrr"r rtr openioBytesIOr#r:ndarrayr$rB ValueErrorformattyperYrvrpredictrkarrayrMrVrSappendryr]rrr;saveseekread)rrrrrrrrrrrr return_typer(rUmaskscutoutsr)rLbios r'removersGH$"4 && 8UZZ 4t3D(EFG D( # # '' 8T" D"** % % (( 8U__T23 l s sT    zz*e,H  $Cg777 OOC 1$ 1& 1EG ??<#?@D F  5-66, (d3%c40v7:F 7|a#G,9'8K' [(zz&!! **,C KKUHHQK 88:= 5,S$7F)#t4F  5s9J88K& K&%K&) F rNFFNF)rs ::  '?N,9,(.&" a 2 2 22 2  2  2j h h 8    h  T(^   xxH" rzz bjj sCc7I1Jx" 8x 8H 87E#s(O77. .1.0$&%)#37$m x+ ,mm),m), m " m k " mmmeCc3./ 0mm C=msmm 5(BJJ &'mq  )* 9: G 34 ?@ G VHHQK sE33AGG