Struct ImageFolder 

pub struct ImageFolder { /* private fields */ }

Legacy ImageFolder dataset for loading images from a directory structure where each subdirectory represents a class

Note: This implementation loads all images into memory at once. For large datasets, consider using OptimizedImageDataset instead.

Implementations

impl ImageFolder

pub fn new<P: AsRef<Path>>(root: P) -> Result<Self>

Create a new ImageFolder dataset

Memory Warning: This loads all images into memory immediately. For datasets larger than a few GB, use OptimizedImageDataset.

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

pub fn get(&self, index: usize) -> Option<(Tensor<f32>, usize)>

pub fn classes(&self) -> &[String]

pub fn class_to_idx(&self) -> &HashMap<String, usize>

Trait Implementations

impl Debug for ImageFolder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.