提交 4f440de2 编写于 作者: X xelmirage 提交者: Waleed

add warning to load_mask of dataset class

use logging.warning() instead of print()
上级 fb76624d
......@@ -378,6 +378,7 @@ class Dataset(object):
"""
# 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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册