dataloader-rs 0.3.0

High-performance DataLoader for Rust with a PyTorch-like interface
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

mod array;
mod batch;
mod collator;
mod ops;
mod transforms;

pub(crate) use batch::{Batch, PyImage, PyImageBatch};
pub(crate) use collator::{ImageCollate, PyImageCollator};
pub(crate) use ops::{Op, Value, apply_all};
pub(crate) use transforms::{PyImageTransform, object_of, value_of};
pub(crate) use transforms::{
    center_crop, compose, crop, decode, hflip, normalize, pad, pad_to, resize, rotate90, to_chw,
    to_float, to_gray, to_hwc, to_rgb, to_rgba, vflip,
};