首页 文章

PyTorch - 加载没有子文件夹的图像

提问于
浏览 870
0

首先,我想说我是PyTorch的新手,所以如果我的问题水平不高,我会在建议中道歉 . 我想知道你是否可以帮助我(我实际上有2个问题) . 背后的故事:我正致力于图像分类 . 我的测试数据根据标签分为子文件夹,我通过DataLoader加载它们 .

第一个问题:1)如果您已经训练了具有特定批量大小的模型,那么使用其他尺寸进行测试是否会影响准确性? 2)有没有办法加载和使用模型与测试数据位于一个文件夹(没有子文件夹) . 据我所知,DataLoader需要子文件夹 .

先感谢您!

1 回答

  • 0
    • 这取决于你是否可以使用像_40728这样的东西,但一般情况下,如果您的网络单独处理批处理项目,它就不会 .

    • 如果查看torch.utils.data.Dataset的文档,您将看到数据集基本上只需要 __len____getitem__ 方法,前者表示数据集包含多少项,后者获取 i 项 - 无论是图像还是标签,图像及其分割蒙版或其他东西 . 没有什么能阻止您编写自定义数据集 . 我建议你看看DatasetFolder的源代码,并根据你的需要进行修改 .

相关问题