add warning to load_mask of dataset class

use logging.warning() instead of print()
nsutphen1 · Feb 10, 2019 · 4f440de · 4f440de
1 parent fb76624
commit 4f440de
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mrcnn/utils.py b/mrcnn/utils.py
@@ -378,6 +378,7 @@ def load_mask(self, image_id):
         """
         # Override this function to load a mask from your dataset.
         # Otherwise, it returns an empty mask.
+        logging.warning("You are using the default load_mask(), maybe you need to define your own one.")
         mask = np.empty([0, 0, 0])
         class_ids = np.empty([0], np.int32)
         return mask, class_ids