burn_dataset/
lib.rs

1//! Dataset loading, splitting, and Burn-compatible batching utilities for CortenForge.
2//!
3//! This crate provides utilities for:
4//! - Loading capture datasets from filesystem
5//! - Train/val splitting with stratification
6//! - Image augmentation pipelines
7//! - Burn-compatible batch iteration
8//! - Warehouse manifest and shard storage
9
10// Module declarations
11pub mod aug;
12pub mod capture;
13pub mod splits;
14pub mod types;
15pub mod validation;
16
17#[cfg(feature = "burn-runtime")]
18pub mod batch;
19#[cfg(feature = "burn-runtime")]
20pub mod warehouse;
21
22// Re-export public API
23pub use aug::{DatasetConfig, TransformPipeline, TransformPipelineBuilder};
24pub use capture::{index_runs, load_run_dataset, load_sample_for_etl, summarize_runs};
25pub use splits::{count_boxes, split_runs, split_runs_stratified};
26pub use types::*;
27pub use validation::{summarize_root_with_thresholds, summarize_with_thresholds, validate_summary};
28
29#[cfg(feature = "burn-runtime")]
30pub use warehouse::{WarehouseLoaders, WarehouseManifest};
31
32#[cfg(feature = "burn-runtime")]
33pub use batch::{build_train_val_iters, BatchIter, BurnBatch};