How do you create an image segmentation data_block with lists of paths?

Part 1 (2018) Beginner (2018)
3

I think i find a way can help your problem



src = (SegmentationItemList.from_folder(path_img)
        .split_none()
#        .split_by_folder(train='img', valid='valid')
       .label_from_func(get_y_fn, classes=codes))

src_valid = (SegmentationItemList.from_folder(valid_path_img)
        .split_none()
       .label_from_func(valid_get_y_fn, classes=codes))

src.valid = src_valid.train

data = (src.transform(get_transforms(), size=size, tfm_y=True)
#         .add_test_folder()   
        .databunch(bs=4)
        .normalize(imagenet_stats))
1 Like