我对此很陌生,只是在学习使用PyTorch!我对如何将给定的数据集拆分为训练和测试(使用图像文件中的标签进行图像分类任务)感到怀疑。
整个数据集共有101个文件夹,每个类别一个。有两个.txt文件,分别是“ train.txt”和“ test.txt”,列出了每个文件夹中的哪些图像分别属于训练和测试数据。
据我了解,数据集目录结构与ImageLoader兼容。但是,我无法弄清楚标签如何与Pytorch中的DataLoader一起使用,以创建训练和测试数据。有人可以帮助我或将我引导到相关资源中我可以阅读的相关信息吗?