test

modules/masking_module.py

@@ -137,6 +137,7 @@ def masking_process(image,obj):
     # task_prompt = '<REGION_TO_SEGMENTATION>'
     # # task_prompt = '<OPEN_VOCABULARY_DETECTION>'
     # print(type(task_prompt),type(obj))
+    print('1')
     image = Image.fromarray(image).convert("RGB")
 
     # results = florence2(image,task_prompt, text_input=obj)
@@ -152,13 +153,17 @@ def masking_process(image,obj):
     # obj = "Tiger"
 
     Florence_results = florence2(image,task_prompt, text_input=obj)
+    print('2')
     SAM_IMAGE_MODEL = load_sam_image_model(device=device)
+    print('3')
     detections = sv.Detections.from_lmm(
                 lmm=sv.LMM.FLORENCE_2,
                 result=Florence_results,
                 resolution_wh=image.size
             )
+    print('4')
     response = run_sam_inference(SAM_IMAGE_MODEL, image, detections)
+    print('5')
     if response['code'] == 400:
       return response
     else: