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

本文解决了一个关于使用ImageFolder加载图片数据时出现的问题。错误原因是指定的路径不正确,导致未能找到预期的图片文件。正确的路径应指向包含实际图片文件的文件夹。

报错如下:

RuntimeError: Found 0 files in subfolders of: /home/A/0
Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,webp

产生原因:路径问题

举例说明: /home/A/0/ 目录下存在若干图片,/home/A/ 目录下只有 0 这一个文件夹。那么ImageFolder的root应该选择哪一个imFolder呢???选择标红一个,产生bug的原因就是选择了 标蓝的一个。

imgFolder=/home/A

imgFolder=/home/A/0

ImageFolder(root=imgFolder, .......)

 

`torchvision.datasets.ImageFolder` 是 PyTorch 中用于加载图像数据集的类,它假定图像数据集按照类别进行文件夹分组,每个文件夹包含属于该类别的图像 [^1]。其语法结构如下: ```python dataset = torchvision.datasets.ImageFolder( root, transform=None, target_transform=None, loader=datasets.folder.default_loader, is_valid_file=None ) ``` 参数说明如下 [^2] [^3]: - `root`:图像文件读取路径,该路径下应包含按类别划分的子文件夹,每个子文件夹中存放该类别的图像。 - `transform`:对图像数据采取的数据增强策略,例如调整大小、裁剪、翻转等。可以使用 `torchvision.transforms` 中的各种变换组合。 - `target_transform`:对标签(label)进行转换的操作。 - `loader`:指定加载图像的函数,默认为 `datasets.folder.default_loader`。 - `is_valid_file`:获取图像路径,检查文件的有效性。 返回值是一个可迭代的数据集对象,可将其放入 `torch.utils.data.DataLoader` 中进行批量加载 [^4]。示例代码如下: ```python import torch from torchvision import datasets, transforms from torch.utils.data import DataLoader # 定义数据转换 transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor() ]) # 加载数据集 dataset = datasets.ImageFolder(root='your_dataset_path', transform=transform) # 创建数据加载器 dataloader = DataLoader(dataset, batch_size=32, shuffle=True) ```
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值