Segmentation Dataset for run length encoding

fastai fastai dev
4

Even easier now that we can specify a mask_opener function, it’s jsut one more line in the data block API:

data = (ImageFileList.from_folder(path_img)
        .label_from_func(get_y_fn)
        .split_by_fname_file('../valid.txt')
        .datasets(SegmentationDataset, classes=codes)
        .set_attr(mask_opener=open_mask_rle)
        .transform(get_transforms(), size=size, tfm_y=True)
        .databunch(bs=bs)
        .normalize(imagenet_stats))
1 Like