udpate: tricks bisenet

inference/tricks.py

@@ -149,12 +149,12 @@ vgg_std = torch.tensor([[[0.229]], [[0.224]], [[0.225]]],
 def load_bisenet():
     bisenet_model = BiSeNet(n_classes=19)
     bisenet_model.load_state_dict(
-        torch.load(make_abs_path("/gavin/datasets/hanbang/79999_iter.pth",), map_location="cpu")
+        torch.load(make_abs_path("./weights/79999_iter.pth",), map_location="cpu")
     )
     bisenet_model.eval()
-    bisenet_model = bisenet_model.cuda(0)
+    bisenet_model = bisenet_model.to(device)
 
-    smooth_mask = SoftErosion(kernel_size=17, threshold=0.9, iterations=7).cuda()
+    smooth_mask = SoftErosion(kernel_size=17, threshold=0.9, iterations=7).to(device)
     print('[Global] bisenet loaded.')
     return bisenet_model, smooth_mask
 

weights/79999_iter.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:468e13ca13a9b43cc0881a9f99083a430e9c0a38abd935431d1c28ee94b26567
+size 53289463