这是我第一次使用PyTorch。我将存储库克隆到googlecolab。在存储库中有一些我想忽略的类(图像文件夹)(假设是文件夹cat)。我怎样才能做到这一点呢?我不想简单地从目录中删除那个类。dataaddress = 'blah/blah' # address where all the classes resides
train_dataset = torchvision.datasets.ImageFolder(dataaddress, transform=train_transfo
我目前正在使用PyTorch在CNN上做实验,我想要完成的任务是对图像进行分类。我知道,使用torchvision.datasets.ImageFolder可以帮助加载所有图像从我的培训文件夹,根据每个子文件夹的名称作为标签。我计划从ImageFolder中随机获取n个图像,但据我所知,ImageFolder没有任何机制可以加载图像的n个数字 of ,其中n是从1到所有可用图像的任意数字。
我怎么能这么做?谢谢你的帮助