又一个抠图Python库

马上注册，结交更多好友，享用更多功能，让你轻松玩转社区。

您需要登录才可以下载或查看，没有账号？立即注册

x

要加载模型才能用
from PIL import Image
import bgRemoverApp

model_name = "U2NET" # U2NET, TRACER
model_path = './ckpt/u2net.pth' # './ckpt/TRACER-Efficient-7.pth'
input_image = Image.open('./input.jpg')

model_pred = bgRemoverApp.load_model(model_name, model_path)
output_image = bgRemoverApp.remove_bg(input_image, model_name, model_pred)
output_image.save('output.png')