Getting closer to understanding what I need from the system

understand.py

@@ -78,6 +78,21 @@ def show_mask_for_number(map_to_use, label_id):
     plt.imshow(visual_mask)
     plt.show()
 
+def show_mask_for_number_over_image(map_to_use, label_id, image_object):
+    """
+    map_to_use: You have to pass in `results["segmentation"]`
+    """
+    if torch.cuda.is_available():
+        mask = (map_to_use.cpu().numpy() == label_id)
+    else:
+        mask = (map_to_use.numpy() == label_id)
+    
+    visual_mask = (mask* 255).astype(np.uint8)
+    visual_mask = Image.fromarray(visual_mask)
+    plt.imshow(image_object)
+    plt.imshow(visual_mask, alpha=0.25)
+    plt.show()
+
 
 def get_coordinates_for_bb_simple(map_to_use, label_id):
     """
@@ -225,6 +240,8 @@ def contour_map(map_to_use, label_id):
         mask = (map_to_use.numpy() == label_id)
     
     visual_mask = (mask* 255).astype(np.uint8)
+    contours, hierarchy = cv.findContours(visual_mask, cv.RETR_TREE, cv.CHAIN_APPROX_SIMPLE)
+    return contours, hierarchy
 
 """
 >>> mask = (results["segmentation"].cpu().numpy() == 1)