Python 在pytorch中创建dataset时，如何取消某些类？

这是我第一次体验PyTorch。我将存储库克隆到google colab。存储库中有一些类（图像文件夹）我想取消（比如文件夹cat）。我怎样才能做到这一点？我不想简单地从目录中删除该类

dataaddress='blah/blah'#所有类所在的地址
train\u dataset=torchvision.datasets.ImageFolder（数据地址，transform=train\u transform）
test\u dataset=torchvision.dataset.ImageFolder（数据地址，transform=eval\u transform）
列_索引=[idx用于范围内的idx（len（列_数据集）），如果idx为%5]
test_索引=[idx代表范围内的idx（len（test_数据集）），如果不是idx%5]
列车数据集=子集（列车数据集、列车索引）
测试数据集=子集（测试数据集、测试索引）

以下是示例代码

filep= os.path.join(root, splitp)

with open(filep, 'r') as splitfile:
  for line in splitfile:
    textlabel=line.split('/')[0].rstrip()
    if textlabel != 'BACKGROUND_Google':
      self.labels.append(textlabel)
      self.labels_idx.append(class_to_idx[textlabel])
      self.images.append(pil_loader(os.path.join(root, line.rstrip())))

---

你找到解决方案了吗？@krivar我发布了它。你能解释一下这段代码的作用以及它是如何解决问题的吗？因此在根目录中有两个文本文件“train.txt”和“test.txt”，它们是用来分割数据的，结构是：accordion/image\u 0001.jpg airplanes/image\u 0150.jpg。上面的“filep”是指向这些文件的路径，“splitp”是接受值“train”或“test”的参数。Rest正在读取这些文件并放弃“后台搜索”。如果您在答案中包含解释，说明代码可以正确格式化，这将更有帮助。