微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

FastAi GrandparentSplitter 文件路径

如何解决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 举报,一经查实,本站将立刻删除。