woo555555的博客

原创 Pytorch中torchvision.datasets.ImageFolder的Found 0 files in subfolders错误

这个错误的主要原因是ImageFolder的路径选择错误应选择数据集图片的上层路径，比如建立一个cat的文件夹存放猫的图片，建立一个dog的文件夹存放狗的图片，这两个文件夹放在train的文件夹内，那么ImageFolder的路径应选择为'.\\train'这样调用 data_image = datasets.ImageFolder(root='.\\train', transform=t...