AugmentationΒΆ

standard_e2e.dataset_utils.augmentation.FrameAugmentation

Abstract base class for per-frame augmentations.

standard_e2e.dataset_utils.augmentation.IdentityFrameAugmentation

Identity augmentation that returns the frame data unchanged.

standard_e2e.dataset_utils.augmentation.MultipleFramesImageAugmentation

Apply an Albumentations pipeline to each camera modality in given frames.

standard_e2e.dataset_utils.augmentation.TrajectoryResampling

Resample past/future/preference trajectories onto target timestamps.