Pytorch加载自己的图片分类数据集(torchvision.datasets.ImageFolder使用详解(数据加载器))
假设数据集文件夹与你的.py文件同路径,如下结构,在这里,我是一个二分类任务的数据集:├─project│├─main.py│├─dataSet││├─train├─ants├─bees││├─val├─ants├─beesfromtorchvision.datasetsimportImageFolderfromtorchvisionimporttransforms#加上transformsnor