add model

model_segvol_single.py

@@ -69,6 +69,7 @@ class SegVolModel(PreTrainedModel):
         if not use_zoom:
             return logits_global_single
         print(logits_global_single.shape)
+        print(torch.unique(logits_global_single))
         if point_prompt_map is not None:
             binary_points = F.interpolate(
                 point_prompt_map.float(),
@@ -79,7 +80,9 @@ class SegVolModel(PreTrainedModel):
                 size=volume_shape, mode='nearest')
         
         min_d, min_h, min_w, max_d, max_h, max_w = logits2roi_coor(self.config.spatial_size, logits_global_single[0][0])
-        assert min_d is None, 'Fail to detect foreground!'
+        if min_d is None:
+            print('Fail to detect foreground!')
+            return logits_global_single
 
         # Crop roi
         image_single_cropped = image[:, :, min_d:max_d+1, min_h:max_h+1, min_w:max_w+1]