Python PyTorch ImageFolder用法及代码示例

本文简要介绍python语言中 torchvision.datasets.ImageFolder 的用法。

用法:

class torchvision.datasets.ImageFolder(root: str, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, loader: Callable[[str], Any] = <function default_loader>, is_valid_file: Optional[Callable[[str], bool]] = None)

参数：

• root(string) -根目录路径。

• transform(可调用的,可选的) -接受 PIL 图像并返回转换版本的函数/转换。例如，transforms.RandomCrop

• target_transform(可调用的,可选的) -接收目标并对其进行转换的函数/转换。

• loader(可调用的,可选的) -在给定路径的情况下加载图像的函数。

• is_valid_file-获取图像文件路径并检查文件是否为有效文件的函数(用于检查损坏的文件)

默认情况下以这种方式排列图像的通用数据加载器：

root/dog/xxx.png
root/dog/xxy.png
root/dog/[...]/xxz.png

root/cat/123.png
root/cat/nsdf3.png
root/cat/[...]/asd932_.png

该类继承自DatasetFolder，因此可以重写相同的方法来自定义数据集。