文章/答案/技术大牛

发布

社区首页 >问答首页 >FastAi GrandparentSplitter文件路径

问FastAi GrandparentSplitter文件路径
EN

Stack Overflow用户

提问于 2021-07-08 16:18:58

回答 1查看 37关注 0票数 1

我正在尝试用FastAi训练一个分割算法。我将训练和验证数据放在不同的文件夹中，所以我计划使用GrandparentSplitter()，但由于某种原因，验证集是空的。

我的文件组织如下：

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

下面是我如何设置我的datablock和dataloader：

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()

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

python

pytorch

fast-ai

回答 1

Stack Overflow用户

发布于 2021-10-04 22:27:12

我不能和祖父母一起工作

just implement my own get_files

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/68297936

复制

相似问题

问FastAi GrandparentSplitter文件路径
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问FastAi GrandparentSplitter文件路径EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问FastAi GrandparentSplitter文件路径
EN