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

【问题标题】：removing sample from a Pytorch dataset object从 Pytorch 数据集对象中删除样本
【发布时间】：2020-11-19 15:30:07
【问题描述】：

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

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

其中有一些我想省略的示例。

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

【问题讨论】：

【解决方案1】：

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

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

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

【讨论】：