Added comment to remember what has to be done

understand.py

@@ -64,7 +64,9 @@ results = processor.post_process_panoptic_segmentation(outputs, target_sizes=[im
 
 
 def show_mask_for_number(map_to_use, label_id):
-
+    """
+    map_to_use: You have to pass in `results["segmentation"]`
+    """
     if torch.cuda.is_available():
         mask = (map_to_use.cpu().numpy() == label_id)
     else:
@@ -77,6 +79,9 @@ def show_mask_for_number(map_to_use, label_id):
 
 
 def get_coordinates_for_bb_simple(map_to_use, label_id):
+    """
+    map_to_use: You have to pass in `results["segmentation"]`
+    """
     if torch.cuda.is_available():
         mask = (map_to_use.cpu().numpy() == label_id)
     else:
@@ -103,6 +108,9 @@ def make_simple_box(left_top, right_bottom, map_size):
 
 
 def test(map_to_use, label_id):
+    """
+    map_to_use: You have to pass in `results["segmentation"]`
+    """
     if torch.cuda.is_available():
         mask = (map_to_use.cpu().numpy() == label_id)
     else: