Блог

Я пытаюсь обучить алгоритм сегментации с помощью FastAi. У меня есть данные обучения и проверки в отдельных папках, поэтому я планировал использовать GrandparentSplitter (), но по какой-то причине набор проверок пуст.

Мои файлы организованы следующим образом:

 Path ---> train ---> images
                ---> masks
     ---> valid ---> images
                ---> masks
 

И вот как я настроил свой блок данных и загрузчик данных:

 codes = np.array(['background', 'prostate'])

def label_func(x): return path/'train/masks'/f'{x.stem}_mask.png'

db = DataBlock(blocks=(ImageBlock(), MaskBlock(codes)),
              splitter=GrandparentSplitter(train_name='train', valid_name='valid'),
              get_items=get_image_files,
              get_y=label_func)

dls = db.dataloaders(path/'train/images', bs=1)
dls.show_batch()
 

Я предполагаю, что что-то не так с тем, как я организовал файлы.

я не мог заставить это работать с бабушкой и дедушкой

просто реализуйте мои собственные get_files