megatron.data.image_folder.DatasetFolder#

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

Bases: VisionDataset

A generic data loader where the samples are arranged in this way: ::: root/class_x/xxx.ext root/class_x/xxy.ext root/class_x/[…]/xxz.ext root/class_y/123.ext root/class_y/nsdf3.ext root/class_y/[…]/asd932_.ext

Parameters:

root (string) – Root directory path.
loader (callable) – A function to load a sample given its path.
extensions (tuple[string]) – A list of allowed extensions. both extensions and is_valid_file should not be passed.
transform (callable, optional) – A function/transform that takes in a sample and returns a transformed version. E.g, transforms.RandomCrop for images.
target_transform (callable, optional) – A function/transform that takes in the target and transforms it.
is_valid_file – A function that takes path of a file and check if the file is a valid file (used to check of corrupt files) both extensions and is_valid_file should not be passed.

__getitem__(index: int) → Tuple[Any, Any]#