add model

model_segvol_single.py

@@ -45,13 +45,10 @@ class SegVolModel(PreTrainedModel):
                 zoomed_image=None, 
                 text_prompt=None, 
                 bbox_prompt=None, 
+                bbox_prompt_map=None,
                 point_prompt=None, 
-                use_zoom=True,
-                **kwargs):
-        if bbox_prompt is not None:
-            bbox_prompt, bbox_prompt_map = bbox_prompt
-        if point_prompt is not None:
-            point_prompt, point_prompt_map = point_prompt
+                point_prompt_map=None,
+                use_zoom=True):
         # print(image.shape, zoomed_image.shape, text_prompt)
         assert image.shape[0] == 1 and zoomed_image.shape[0] == 1, 'batch size should be 1'
         # print(bbox_prompt.shape, bbox_prompt_map.shape, point_prompt[0].shape, point_prompt[1].shape, point_prompt_map.shape)