Class ImageFolder##
class torchvision.datasets.ImageFolder(root: ~typing.Union[str, ~pathlib.Path], transform: typing.Optional[typing.Callable] = None, target_transform: typing.Optional[typing.Callable] = None, loader: ~typing.Callable[[str], ~typing.Any] = , is_valid_file: typing.Optional[typing.Callable[[str], bool]] = None, allow_empty: bool = False)
就该类输出的格式
A generic data loader where the images are arranged in this way by default:
一个通用的默认排列方式的图像数据加载器：
root/dog/xxx.png
root/dog/xxy.png
root/dog/[...]/xxz.png
root/cat/123.png
root/cat/nsdf3.png
root/cat/[...]/asd932_.png
parameters;
This class inherits from DatasetFolder so the same methods can be overridden to customize the dataset.
- Parameters:
-
root (str or pathlib.Path) -- Root directory path.
-
transform (callable, optional) -- A function/transform that takes in a PIL image or torch.Tensor, depends on the given loader, and returns a transformed version. E.g, transforms.RandomCrop
-
target_transform (callable, optional) -- A function/transform that takes in the target and transforms it.
-
loader (callable, optional) -- A function to load an image given its path.
-
is_valid_file (callable, optional) -- A function that takes path of an Image file and check if the file is a valid file (used to check of corrupt files)
-
allow_empty -- If True, empty folders are considered to be valid classes. An error is raised on empty folders if False (default).
-
参数
- root （数据集根目录路径）
- transform （可选参数；回调函数。就是torchvision中的transform模块中的类对象）。ImageFolder类对象并不会立即执行，而是在你需要该数据时 才会对root路径下的数据进行执行该函数。