FrameLoader

class standard_e2e.dataset_utils.frame_loader.FrameLoader(frame_name, required_modalities, frame_selector, processed_data_path=None)

Bases: object

Loads individual frames from processed data with selection logic.

Parameters:

• frame_name (str)

• required_modalities (list[Modality])

• frame_selector (FrameSelector)

• processed_data_path (str | None)

load_frame(current_frame_idx, index_data, modality_defaults=None)

Load and materialize a frame selected relative to current_frame_idx.

Uses the configured FrameSelector to map the requested index to the required frame, loads it from disk, attaches modality defaults, and computes timestamp_diff vs. the current frame.

Returns a mapping of frame_name -> TransformedFrameData for easy merging across multiple loaders.

Return type:

dict[str, TransformedFrameData]

Parameters:

• current_frame_idx (int)

• index_data (DataFrame)

• modality_defaults (dict[Modality, ModalityDefaults] | None)

property processed_data_path: str | None

Get the processed data path.

set_index_data(index_data)

Set the index data for the frame selector.

Parameters:

index_data (DataFrame)

set_processed_data_path(processed_data_path)

Set the processed data path.

Parameters:

processed_data_path (str)