Crate shrew_data 

shrew-data

Data loading, datasets, and batching for Shrew.

This crate provides:

• Dataset trait — unified interface for any dataset
• DataLoader — batching, shuffling, parallel iteration over a Dataset
• AsyncDataLoader — prefetching data loader with background workers

Re-exports

pub use async_loader::AsyncDataLoader;

pub use async_loader::AsyncDataLoaderConfig;

pub use async_loader::Batch;

pub use async_loader::PrefetchIterator;

pub use augment::ColorJitter;

pub use augment::RandomCrop;

pub use augment::RandomErasing;

pub use augment::RandomHorizontalFlip;

pub use augment::RandomNoise;

pub use augment::RandomVerticalFlip;

pub use combinators::train_test_split;

pub use combinators::ConcatDataset;

pub use combinators::MapDataset;

pub use combinators::SubsetDataset;

pub use combinators::VecDataset;

pub use csv_dataset::CsvConfig;

pub use csv_dataset::CsvDataset;

pub use dataset::Dataset;

pub use dataset::Sample;

pub use loader::DataLoader;

pub use loader::DataLoaderConfig;

pub use mnist::MnistDataset;

pub use transform::Transform;

Modules

async_loader

augment

combinators

csv_dataset

dataset

image_folder

loader

mnist

transform

Enums

ImageFolderError

Errors from the ImageFolder dataset.