A generic data loader for images stored in folders.
Details for more information.
image_folder_dataset( root, transform = NULL, target_transform = NULL, loader = NULL, is_valid_file = NULL )
Root directory path.
A function/transform that takes in an PIL image and returns
a transformed version. E.g,
A function/transform that takes in the target and transforms it.
A function to load an image given its path.
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
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