yuxindu
/

segvol

Image Segmentation

feature-extraction

Model card Files Files and versions

yuxin commited on Apr 11, 2024

Commit

c4cc548

·

1 Parent(s): c995392

add model

Files changed (1) hide show

model_segvol_single.py +1 -1

model_segvol_single.py CHANGED Viewed

@@ -247,7 +247,7 @@ def select_points(preds, num_positive_extra=4, num_negative_extra=0, fix_extra_p
         points = torch.cat((points, ignore_point), dim=0)
         labels = torch.cat((labels, torch.tensor([-1]).reshape(1)))
-    return (points, labels)
 # SegVol
 import torch

         points = torch.cat((points, ignore_point), dim=0)
         labels = torch.cat((labels, torch.tensor([-1]).reshape(1)))
+    return points, labels
 # SegVol
 import torch