Struct VideoFolder 

pub struct VideoFolder { /* private fields */ }

Video dataset for loading video files from directories

Implementations

impl VideoFolder

pub fn new<P: AsRef<Path>>(root: P) -> Result<Self>

Create a new video folder dataset

pub fn with_max_frames(self, max_frames: usize) -> Self

Set maximum number of frames to extract

pub fn with_frame_rate(self, fps: f32) -> Self

Set target frame rate for extraction

pub fn with_transform<T>(self, transform: T) -> Self

where T: Transform<VideoFrames, Output = Tensor<f32>> + 'static,

Set transform to apply to video frames

pub fn classes(&self) -> &[String]

Get class names

pub fn root(&self) -> &Path

Get the root directory path

pub fn num_samples(&self) -> usize

Get the number of samples

Trait Implementations

impl Dataset for VideoFolder

type Item = (Tensor, usize)

The type of items returned by the dataset

fn len(&self) -> usize

Returns the number of items in the dataset

fn get(&self, index: usize) -> Result<Self::Item>

Get a single item from the dataset

fn is_empty(&self) -> bool

Returns true if the dataset is empty