A generic data loader for images stored in folders.
See Details
for more information.
image_folder_dataset( root, transform = NULL, target_transform = NULL, loader = NULL, is_valid_file = NULL )
root | Root directory path. |
---|---|
transform | A function/transform that takes in an PIL image and returns
a transformed version. E.g, |
target_transform | A function/transform that takes in the target and transforms it. |
loader | 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) |
This function assumes that the images for each class are contained
in subdirectories of root
. The names of these subdirectories are stored
in the classes
attribute of the returned object.
An example folder structure might look as follows:
root/dog/xxx.png root/dog/xxy.png root/dog/xxz.png root/cat/123.png root/cat/nsdf3.png root/cat/asd932_.png