Struct Feature

pub struct Feature {
    pub dtype: DType,
    pub shape: Vec<usize>,
    pub names: Option<Names>,
}

Feature definition for a LeRobot dataset.

Each feature represents a data stream recorded during an episode, of a specific data type (dtype) and dimensionality (shape).

For example, a shape of [3, 224, 224] for a DType::Image feature denotes a 3-channel (e.g. RGB) image with a height and width of 224 pixels each.

Fields

dtype: DType

shape: Vec<usize>

names: Option<Names>

Implementations

impl Feature

pub fn channel_dim(&self) -> usize

Get the channel dimension for this Feature.

Returns the number of channels in the feature’s data representation.

Note

This is primarily intended for DType::Image and DType::Video features, where it represents color channels (e.g., 3 for RGB, 4 for RGBA). For other feature types, this function returns the size of the last dimension from the feature’s shape.

Trait Implementations

impl Clone for Feature

fn clone(&self) -> Feature

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Feature

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Feature

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for Feature

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.