megatron.data.image_folder.ImageFolder#

class megatron.data.image_folder.ImageFolder(*args: Any, **kwargs: Any)#

A generic data loader where the images are arranged in this way: ::: 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:

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)