BiRefNet_Backgroun_Removal

Running

ZhengPeng7 commited on Jul 10, 2024

Commit

840b51f

1 Parent(s): 0851b27

Update weights.

Files changed (4) hide show

materials/examples/1#Accessories#5#Jewelry#12836147075_09291d1d24_o.jpg → BiRefNet-massive-epoch_260.pth RENAMED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7ac3b5f61ca2d282901d0932921f500b65344a0f0e91b3dc81599cf39c1d7262
-size 985808

 version https://git-lfs.github.com/spec/v1
+oid sha256:92bb252307a56c9bbf95a258a96f60e96a8ccbb8d45275de0b8c76dbda0cd42a
+size 885038517

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ class ImagePreprocessor():
 model = BiRefNet(bb_pretrained=False)
-state_dict = ['./BiRefNet-DIS-epoch_590.pth', 'BiRefNet-massive-epoch_240.pth'][0]
 if os.path.exists(state_dict):
     birefnet_dict = torch.load(state_dict, map_location="cpu")
     unwanted_prefix = '_orig_mod.'

 model = BiRefNet(bb_pretrained=False)
+state_dict = ['./BiRefNet-DIS-epoch_590.pth', 'BiRefNet-massive-epoch_240.pth', 'BiRefNet-massive-epoch_260.pth'][-1]
 if os.path.exists(state_dict):
     birefnet_dict = torch.load(state_dict, map_location="cpu")
     unwanted_prefix = '_orig_mod.'

materials/examples/20#Sports#8#Racket#4827171149_3140bffe12_o.jpg DELETED Viewed

materials/examples/4#Architecture#3#Crack#3551574032_17dd106d31_o.jpg DELETED Viewed