Max Meyer commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -33,8 +33,7 @@ model = model.BEN_Base().to(device).eval() #init pipeline
|
|
| 33 |
|
| 34 |
model.loadcheckpoints("./BEN_Base.pth")
|
| 35 |
image = Image.open(file)
|
| 36 |
-
|
| 37 |
-
mask, foreground = model.inference(image)
|
| 38 |
|
| 39 |
mask.save("./mask.png")
|
| 40 |
foreground.save("./foreground.png")
|
|
|
|
| 33 |
|
| 34 |
model.loadcheckpoints("./BEN_Base.pth")
|
| 35 |
image = Image.open(file)
|
| 36 |
+
mask, foreground = model.inference(image)
|
|
|
|
| 37 |
|
| 38 |
mask.save("./mask.png")
|
| 39 |
foreground.save("./foreground.png")
|