ImageFolder

class ImageFolder(root, check_valid_func=None, class_name=False)

ImageFolder is a class for loading image data and labels from a organized folder.

The folder is expected to be organized as followed: root/cls/xxx.img_ext

Labels are indices of sorted classes in the root directory.

Parameters:

• root (str) – root directory of an image folder.

• loader – a function used to load image from path, if None, default function that loads images with PIL will be called.

• check_valid_func – a function used to check if files in folder are expected image files, if None, default function that checks file extensions will be called.

• class_name (bool) – if True, return class name instead of class index.