【pytorch | torchvision | datasets】ImageFolder（）类

拾贰_C2025-12-19 9:53

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路径下的数据进行执行该函数。