Spaces:
Running
Running
dennistrujillo
commited on
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)
|