deep-learning - pytorch ImageFolder 数据大小与原始 imgs 编号不匹配?
问题描述
我在 train 文件夹中有 21,522 个项目,总计 609.7 MB
train
-1(3600 png)
-2(3600 png)
.
.
-6
train_trans = transforms.Compose([
vision.transforms.Resize(target_size),
vision.transforms.RandomHorizontalFlip(),
vision.transforms.RandomRotation(20),
#CIFAR10Policy(),
vision.transforms.ToTensor(),
vision.transforms.Normalize(
[0.485, 0.456, 0.406],
[0.229, 0.224, 0.225])
])
valid_trans = transforms.Compose([
vision.transforms.Resize(target_size),
vision.transforms.RandomResizedCrop(target_size, scale=(0.8,1.0)),
vision.transforms.RandomHorizontalFlip(),
vision.transforms.ToTensor(),
vision.transforms.Normalize(
[0.485, 0.456, 0.406],
[0.229, 0.224, 0.225])
])
test_trans = transforms.Compose([
vision.transforms.Resize((size,size)),
vision.transforms.RandomResizedCrop(target_size, scale=(0.8,1.0)),
vision.transforms.ToTensor(),
vision.transforms.Normalize(
[0.485, 0.456, 0.406],
[0.229, 0.224, 0.225])
])
但是当我加载数据集时
train_dataset = torchvision.datasets.ImageFolder(root= path + 'train/',transform=trans)
不知何故
打印(len(train_dataset)) = 4109
不知何故,它不会加载整个图像文件
解决方案
如果要访问给定文件夹中的所有图像,则给定的路径应该在上一级。
例如,当您将路径指定为 时root= path + 'train/'
,则ImageFolder
仅查看文件夹中的../train/
文件夹,而不查看该文件夹中的图像文件。因此,要么在 train 文件夹中的文件夹中组织文件,要么确保根文件夹中只有 train 文件夹并提供根文件夹的路径。
推荐阅读
- reactjs - 两个或多个 Route 可以渲染同一个组件吗?
- java - 从行具有特定值的表中检索 Firebase 中的数据
- javascript - 反应原生构建失败的Android
- javascript - 指针在 JavaScript 中是如何工作的?
- javascript - Javascript/jQuery:第一次 AJAX 调用成功时开始“闪烁”,下一次停止
- pointers - 如何递归地将 2 个排序的链表合并到一个新的排序列表中?
- python - Python3 - 识别对象类型的问题
- sql - 在 SQL 表上创建统计信息之前检查是否存在具有相同列的统计信息
- php - 从 PHP 字符串中获取值
- php - 生成HTML转PDF,并将生成的PDF上传到存储(laravel)(Dompdf库)