dennistrujillo commited on
Commit
4b69b97
·
verified ·
1 Parent(s): af6805d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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)