Crate burn_dataset

Crate burn_dataset 

Source
Expand description

Dataset loading, splitting, and Burn-compatible batching utilities for CortenForge.

This crate provides utilities for:

  • Loading capture datasets from filesystem
  • Train/val splitting with stratification
  • Image augmentation pipelines
  • Burn-compatible batch iteration
  • Warehouse manifest and shard storage

Re-exports§

pub use aug::DatasetConfig;
pub use aug::TransformPipeline;
pub use aug::TransformPipelineBuilder;
pub use capture::index_runs;
pub use capture::load_run_dataset;
pub use capture::load_sample_for_etl;
pub use capture::summarize_runs;
pub use splits::count_boxes;
pub use splits::split_runs;
pub use splits::split_runs_stratified;
pub use validation::summarize_root_with_thresholds;
pub use validation::summarize_with_thresholds;
pub use validation::validate_summary;
pub use warehouse::WarehouseLoaders;
pub use warehouse::WarehouseManifest;
pub use batch::build_train_val_iters;
pub use batch::BatchIter;
pub use batch::BurnBatch;
pub use types::*;

Modules§

aug
Image augmentation and transformation pipeline.
batch
Batch iteration for training and validation.
capture
Loading and indexing capture dataset files.
splits
Train/validation splitting logic with stratification support.
types
Core types, error definitions, and data structures for burn_dataset.
validation
Dataset validation and quality checks.
warehouse
Warehouse manifest and shard storage backends.