Pytorch

Question

继上次之后，我现在正在尝试实现 torchvision.datasets.ImageFolder class 的子 class。下面的代码returns一个错误（"name 'default_loader' is not defined"），我想不通为什么。你能帮帮我吗？

class ExtendingImageFolder(torchvision.datasets.ImageFolder)
   def __init__(self,root,transform=None, target_transform=None,loader=default_loader):
       super().__init__(root,transform,target_transform,loader)

当我删除“None”和“default_loader”时，这样写；

    class ExtendingImageFolder(torchvision.datasets.ImageFolder)
   def __init__(self,root,transform, target_transform,loader):
       super().__init__(root,transform,target_transform,loader)

我在尝试创建此 class 的实例时收到缺少输入参数的错误，例如：

JJ=ExtendingImageFolder(root='C:/',transform=transform)

我做错了什么？

提前致谢！

Answer 1

default_loader() 是 torchvision/datasets/folder.py 中定义的函数，以及 ImageFolder 和其他基于文件夹的数据集助手。

然而，它不会在 torchvision/datasets/__init__.py 中导出（与 ImageFolder 不同）。您仍然可以直接使用“from torchvision.datasets.folder import default_loader”导入它 - 这应该可以解决您的导入错误。

Pytorch - torchvision.dataset.ImageFolder 的子类 - 导入错误

Pytorch - Subclasses of torchvision.dataset.ImageFolder - Import Error

python

python-import