def __init__(self, root_dir, split_name, transform):
    super(ISSDataset, self).__init__()
    self.root_dir = root_dir
    self.split_name = split_name
    self.transform = transform
    self._img_dir = path.join(root_dir, ISSDataset._IMG_DIR)
    self._msk_dir = path.join(root_dir, ISSDataset._MSK_DIR)
    self._lst_dir = path.join(root_dir, ISSDataset._LST_DIR)
    for d in (self._img_dir, self._msk_dir, self._lst_dir):
        if (not path.isdir(d)):
            raise IOError('Dataset sub-folder {} does not exist'.format(d))
    (self._meta, self._images) = self._load_split()
def __init__(self, root_dir, split_name, transform):
    super(ISSDataset, self).__init__()
    self.root_dir = root_dir
    self.split_name = split_name
    self.transform = transform
    self._img_dir = path.join(root_dir, ISSDataset._IMG_DIR)
    self._msk_dir = path.join(root_dir, ISSDataset._MSK_DIR)
    self._lst_dir = path.join(root_dir, ISSDataset._LST_DIR)
    for d in (self._img_dir, self._msk_dir, self._lst_dir):
        if (not path.isdir(d)):
            raise IOError('Dataset sub-folder {} does not exist'.format(d))
    (self._meta, self._images) = self._load_split()