Spaces:

esun-choi
/

InvoHide_Inisw_teem8

Sleeping

esun-choi commited on Dec 19, 2023

Commit

b5283fc

1 Parent(s): 022722e

Update seg2.py

Files changed (1) hide show

seg2.py CHANGED Viewed

@@ -10,7 +10,7 @@ import cv2
 T2=transforms.ToPILImage()
-weights="/home/ksyint/other1213/craft_ku/weights/dinov2.pt"
 device=torch.device("cpu")
 model = Segmentor(device,1,backbone = 'dinov2_b',head="conv")
 model.load_state_dict(torch.load(weights,map_location="cpu"))

 T2=transforms.ToPILImage()
+weights="./weights/dinov2.pt"
 device=torch.device("cpu")
 model = Segmentor(device,1,backbone = 'dinov2_b',head="conv")
 model.load_state_dict(torch.load(weights,map_location="cpu"))