yuxin
commited on
Commit
•
6dad4ed
1
Parent(s):
d238b7e
add model
Browse files- model_segvol_single.py +1 -1
model_segvol_single.py
CHANGED
@@ -32,7 +32,7 @@ class SegVolModel(PreTrainedModel):
|
|
32 |
bbox_prompt, bbox_prompt_map = bbox_prompt
|
33 |
point_prompt, point_prompt_map = point_prompt
|
34 |
print(image.shape, zoomed_image.shape, text_prompt)
|
35 |
-
print(bbox_prompt.shape, bbox_prompt_map
|
36 |
# test mode
|
37 |
if self.config.test_mode:
|
38 |
return
|
|
|
32 |
bbox_prompt, bbox_prompt_map = bbox_prompt
|
33 |
point_prompt, point_prompt_map = point_prompt
|
34 |
print(image.shape, zoomed_image.shape, text_prompt)
|
35 |
+
print(bbox_prompt.shape, bbox_prompt_map.shape, point_prompt[0].shape, point_prompt[1].shape, point_prompt_map.shape)
|
36 |
# test mode
|
37 |
if self.config.test_mode:
|
38 |
return
|