gavinyuan commited on
Commit
ee36df0
1 Parent(s): 2b471ce

udpate: tricks bisenet

Browse files
Files changed (1) hide show
  1. 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("./weights/79999_iter.pth",), map_location="cpu")
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)