add model

model_segvol_single.py

@@ -45,6 +45,7 @@ class SegVolModel(PreTrainedModel):
         point_prompt, point_prompt_map = point_prompt
         print(image.shape, zoomed_image.shape, text_prompt)
         print(bbox_prompt.shape, bbox_prompt_map.shape, point_prompt[0].shape, point_prompt[1].shape, point_prompt_map.shape)
+        volume_shape = image[0][0].shape
         # test mode
         if self.config.test_mode:
             with torch.no_grad():
@@ -53,6 +54,10 @@ class SegVolModel(PreTrainedModel):
                                                     boxes=bbox_prompt, 
                                                     points=point_prompt)
             print(logits_global_single.shape)
+            logits_global_single = F.interpolate(
+                logits_global_single.cpu(),
+                size=volume_shape, mode='nearest')[0][0]
+            print(logits_global_single.shape)
             return
         else:
             print('unsupport training mode now')