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

Fix path parsing error in previous commit

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