print mask info

Author: abrichr

task_adapter/sam/tasks/inference_sam_m2m_auto.py

@@ -45,6 +45,8 @@ def inference_sam_m2m_auto(model, image, text_size, label_mode='1', alpha=0.1, a
     mask_map = np.zeros(image_ori.shape, dtype=np.uint8)    
     for i, ann in enumerate(sorted_anns):
         mask = ann['segmentation']
+        print(f"{type(mask)=}")
+        print(f"{mask.dtype=} {mask.shape=} {mask.min()=} {mask.max()=}")
         color_mask = np.random.random((1, 3)).tolist()[0]
         # color_mask = [int(c*255) for c in color_mask]
         demo = visual.draw_binary_mask_with_number(mask, text=str(label), label_mode=label_mode, alpha=alpha, anno_mode=anno_mode)
@@ -100,4 +102,4 @@ def show_anns(anns):
         color_mask = np.random.random((1, 3)).tolist()[0]
         for i in range(3):
             img[:,:,i] = color_mask[i]
-        ax.imshow(np.dstack((img, m*0.35)))
+        ax.imshow(np.dstack((img, m*0.35)))