image_folder_dataset.hpp File Reference

Directory-of-class-subfolders image Dataset, mirroring torchvision's ImageFolder. More...

#include <string>
#include <vector>
#include "pulsatrix/dataset.hpp"
#include "pulsatrix/device_backend.hpp"

Include dependency graph for image_folder_dataset.hpp:

Go to the source code of this file.

Classes
class	pulsatrix::ImageFolderDataset
	Dataset over a directory tree of the form root_dir/<class_name>/<image_file>, mirroring torchvision's ImageFolder convention. Class names are the sorted subdirectory names; each class's label is its index in that sorted order. Images are decoded lazily (per get() call) via ImageDecoder. More...

Namespaces
namespace	pulsatrix

Detailed Description

Directory-of-class-subfolders image Dataset, mirroring torchvision's ImageFolder.