U {B$g@s.ddlmZmZddlmZGdddZdS))remove new_session)timing_decoratorc@s(eZdZd ddZedd ddZdS) Removebgu2netcCst||_dS)z! name: rembg N)rsession)selfnamer X/apdcephfs_cq5/share_300600172/huiwenshi/3d_services/repos/hunyuan3d-lite/infer/rembg.py__init__szRemovebg.__init__zremove backgroundFcCs0|jdkr|r|d}n|St||jd}|S)z inputs: rgb_img: PIL.Image, with RGB mode expected force: bool, input is RGBA mode return: rgba_img: PIL.Image with RGBA mode RGBARGB)r)modeconvertrr)r Zrgb_imgforceZrgba_imgr r r __call__ s  zRemovebg.__call__N)r)F)__name__ __module__ __qualname__r rrr r r r rs rN)rembgrrutilsrrr r r r s