我目前正在使用PyTorch进行CNN实验，我希望模型完成的任务是对图像进行分类。
我了解根据每个子文件夹的名称作为标签，使用torchvision.datasets.ImageFolder可以帮助从训练文件夹中加载所有图像。
我计划仅从ImageFolder中随机获取n张图片，但据我所知，ImageFolder没有机制来加载n张图片 随机图像，其中n是从1到所有可用图像的任意数字。
我该怎么做？感谢您的帮助
答案 0 :(得分：0)
您可以使用PyTorch的{{1}}类创建ImageFolder的子集。我们可以根据需要使用numpy或其他方法生成随机索引。
Subset