pytorch torchvision.datasets.ImageFolder

API

CLASS torchvision.datasets.ImageFolder(root, transform=None, target_transform=None, loader=<function default_loader>, is_valid_file=None)

存放格式

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

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

参数	描述
root (string)	Root directory path.
transform (callable, optional)	A function/transform that takes in an PIL image 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	A function that takes path of an Image file and check if the file is a valid file (used to check of corrupt files)

参考：
https://pytorch.org/docs/stable/torchvision/datasets.html#torchvision.datasets.ImageFolder