init project

modules/pe3r/demo.py

@@ -257,6 +257,8 @@ def get_mask_from_img_sam1(mobilesamv2, yolov8, sam1_image, yolov8_image, origin
     input_image = mobilesamv2.preprocess(sam1_image)
     image_embedding = mobilesamv2.image_encoder(input_image)
 
+    print(image_embedding)
+
     image_embedding=torch.repeat_interleave(image_embedding, 320, dim=0)
     prompt_embedding=mobilesamv2.prompt_encoder.get_dense_pe()
     prompt_embedding=torch.repeat_interleave(prompt_embedding, 320, dim=0)