add model

model_segvol_single.py

@@ -98,6 +98,7 @@ class SegVolProcessor():
         return item
 
     def point_prompt_b(self, label_single_resize):
+        print(label_single_resize.shape, torch.unique(label_single_resize))
         point, point_label = select_points(label_single_resize, num_positive_extra=3, num_negative_extra=3)
         points_single = (point.unsqueeze(0).float().cuda(), point_label.unsqueeze(0).float().cuda()) 
         binary_points_resize = build_binary_points(point, point_label, label_single_resize.shape).unsqueeze(0).unsqueeze(0)