又一个抠图Python库
要加载模型才能用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')
页:
[1]