Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,6 +34,11 @@ def medsam_inference(medsam_model, img_embed, box_1024, H, W):
|
|
| 34 |
boxes=box_torch,
|
| 35 |
masks=None,
|
| 36 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
low_res_logits, _ = medsam_model.mask_decoder(
|
| 38 |
image_embeddings=img_embed, # (B, 256, 64, 64)
|
| 39 |
image_pe=medsam_model.prompt_encoder.get_dense_pe(), # (1, 256, 64, 64)
|
|
|
|
| 34 |
boxes=box_torch,
|
| 35 |
masks=None,
|
| 36 |
)
|
| 37 |
+
|
| 38 |
+
#print shapes of tensors
|
| 39 |
+
print("Shape of sparse_embeddings:", sparse_embeddings.shape)
|
| 40 |
+
print("Shape of dense_embeddings:", dense_embeddings.shape)
|
| 41 |
+
|
| 42 |
low_res_logits, _ = medsam_model.mask_decoder(
|
| 43 |
image_embeddings=img_embed, # (B, 256, 64, 64)
|
| 44 |
image_pe=medsam_model.prompt_encoder.get_dense_pe(), # (1, 256, 64, 64)
|