Spaces:
Running
Running
gavinyuan
commited on
Commit
•
ee36df0
1
Parent(s):
2b471ce
udpate: tricks bisenet
Browse files- inference/tricks.py +1 -1
inference/tricks.py
CHANGED
@@ -149,7 +149,7 @@ vgg_std = torch.tensor([[[0.229]], [[0.224]], [[0.225]]],
|
|
149 |
def load_bisenet():
|
150 |
bisenet_model = BiSeNet(n_classes=19)
|
151 |
bisenet_model.load_state_dict(
|
152 |
-
torch.load(make_abs_path("
|
153 |
)
|
154 |
bisenet_model.eval()
|
155 |
bisenet_model = bisenet_model.to(device)
|
|
|
149 |
def load_bisenet():
|
150 |
bisenet_model = BiSeNet(n_classes=19)
|
151 |
bisenet_model.load_state_dict(
|
152 |
+
torch.load(make_abs_path("../weights/79999_iter.pth",), map_location="cpu")
|
153 |
)
|
154 |
bisenet_model.eval()
|
155 |
bisenet_model = bisenet_model.to(device)
|