FastAi GrandparentSplitter 文件路径

问题描述

我正在尝试使用 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()

我假设我组织文件的方式有问题。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

fast-ai python pytorch