Struct burn_core::data::dataset::vision::ImageFolderDataset

pub struct ImageFolderDataset { /* private fields */ }

A generic dataset to load images from disk.

Implementations

impl ImageFolderDataset

pub fn new_classification<P>( root: P ) -> Result<ImageFolderDataset, ImageLoaderError>

where P: AsRef<Path>,

Create an image classification dataset from the root folder.

Arguments

• root - Dataset root folder.

Returns

A new dataset instance.

pub fn new_classification_with<P, S>( root: P, extensions: &[S] ) -> Result<ImageFolderDataset, ImageLoaderError>

where P: AsRef<Path>, S: AsRef<str>,

Create an image classification dataset from the root folder. The included images are filtered based on the provided extensions.

Arguments

• root - Dataset root folder.
• extensions - List of allowed extensions.

Returns

A new dataset instance.

pub fn new_classification_with_items<P, S>( items: Vec<(P, String)>, classes: &[S] ) -> Result<ImageFolderDataset, ImageLoaderError>

where P: AsRef<Path>, S: AsRef<str>,

Create an image classification dataset with the specified items.

Arguments

• items - List of dataset items, each item represented by a tuple (image path, label).
• classes - Dataset class names.

Returns

A new dataset instance.

pub fn new_multilabel_classification_with_items<P, S>( items: Vec<(P, Vec<String>)>, classes: &[S] ) -> Result<ImageFolderDataset, ImageLoaderError>

where P: AsRef<Path>, S: AsRef<str>,

Create a multi-label image classification dataset with the specified items.

Arguments

• items - List of dataset items, each item represented by a tuple (image path, labels).
• classes - Dataset class names.

Returns

A new dataset instance.

Trait Implementations

impl Dataset<ImageDatasetItem> for ImageFolderDataset

fn get(&self, index: usize) -> Option<ImageDatasetItem>

Gets the item at the given index.

fn len(&self) -> usize

Gets the number of items in the dataset.

fn is_empty(&self) -> bool

Checks if the dataset is empty.

fn iter(&self) -> DatasetIterator<'_, I>

where Self: Sized,

Returns an iterator over the dataset.