提交 a9e598f9 编写于 作者: W Waleed Abdulla

Fix path parsing error in previous commit

上级 ea63f95f
......@@ -226,7 +226,7 @@ class NucleusDataset(utils.Dataset):
"""
info = self.image_info[image_id]
# Get mask directory from image path
mask_dir = os.path.join(os.path.dirname(os.path.dirname(path)), "masks")
mask_dir = os.path.join(os.path.dirname(os.path.dirname(info['path'])), "masks")
# Read mask files from .png image
mask = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册