从Pytorch数据集对象中删除样本

Question

我正在使用以下方式将图像加载到pytorch：

train_data = torchvision.datasets.ImageFolder(train_dir, transform=transform)

其中有一些样本我想略去

我可以从加载程序中删除某些图像吗？（按索引？按名称？） Thx

Answer 1

您可以修改ImageFolder的imgs属性，如下所示：

train_data.imgs.remove((PATH, INDEX))

其中PATH应替换为要删除的映像的路径，而INDEX应替换为相应的类索引（从0到您拥有的类数）。 如果您对INDEX不确定，可以打印train_data.imgs以获取所有图像路径及其元组类的列表