PramaLLC
/

BEN

Image Segmentation

background-remove

mask-generation

Dichotomous image segmentation

background remove

remove background

Model card Files Files and versions

Max Meyer commited on Nov 13, 2024

Commit

bbb0aa9

·

verified ·

1 Parent(s): 7ed52c3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ import torch
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-file = "./image.jpg" # input image
 model = model.BEN_Base().to(device).eval() #init pipeline

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+file = "./image.png" # input image
 model = model.BEN_Base().to(device).eval() #init pipeline