如何使用.txt文件中的给定标签在Pytorch中将文件夹分为测试和培训数据?

时间:2019-05-29 14:26:27

标签: label pytorch training-data test-data

我对此很陌生,只是在学习使用PyTorch!我对如何将给定的数据集拆分为训练和测试(使用图像文件中的标签进行图像分类任务)感到怀疑。

整个数据集共有101个文件夹,每个类别一个。有两个.txt文件,分别是“ train.txt”和“ test.txt”,列出了每个文件夹中的哪些图像分别属于训练和测试数据。

据我了解,数据集目录结构与ImageLoader兼容。但是,我无法弄清楚标签如何与Pytorch中的DataLoader一起使用,以创建训练和测试数据。有人可以帮助我或将我引导到相关资源中我可以阅读的相关信息吗?

0 个答案:

没有答案