我确实有一个图像多分类问题,我所有的图像都存储在一个文件夹中,每个图像的标签都在其文件名中。
我是PyTorch的新手,想知道为什么(据我所知)只有一种方法,比如ImageFolder()
来构建Dataset
?在我看来,根据ImageFolder()
及其预定义的train和test文件夹来重构我的图像似乎很麻烦。
把所有的图片都放在一个文件夹里,并把标签放在文件名里,这样的结构合理吗?如果是这样,为什么没有像ImageFromOneFolder()
这样的Dataset
方法呢?
我想我们要做一个切割数据集。
谢谢你的帮助
1条答案
按热度按时间ut6juiuv1#
自定义数据集可以工作:
字符串