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
|