Views
No views yet
.pth and ONNX formats.

1from PIL import Image
2from utilities import preprocess_image, postprocess_image, load_model
3
4model = load_model("model.pth")
5image = Image.open("example_input.jpg")
6mask = model.predict(image)
7image.putalpha(mask)
8image.save("output.png")