PyTorch Pascal-5i数据集读取器
一个有效的Python实现,用于构建pascal-5i数据集。
PASCAL-5i是很少使用的分割文献中经常使用的数据集。 然而,似乎没有为工作的实施可在网上:
以一种相当混乱的方式编写(至少对我而言)。
是我发现的唯一其他开源实现。 但是,该代码无法正常工作,并且存在一些未解决的GitHub问题。
因此,我想为我自己和对使用pascal-5i感兴趣的任何人实现此工具。 这实际上是一个PyTorch数据集读取器。 i折叠是在数据集初始化阶段完成的。 因此,它在训练/推理过程中没有任何开销。
用法
在examples.ipynb中查看用法示例!
如果要将数据集下载到/data ,则可以运行ipython笔记本,并将download=False更改为download=True 。 使用torchvision实用程序功能, 和将自动下载到
1