Expand description
Image augmentation for visual RL (DrQ-v2 style random shift).
Provides the ImageAugmentation trait and concrete implementations
for composable, reproducible image augmentations on flat (B, C, H, W) arrays.
Structs§
- Random
Shift - DrQ-v2 random shift augmentation.
Traits§
- Image
Augmentation - Trait for composable image augmentations.
Functions§
- random_
shift_ batch - Random shift: pad image with zeros, then crop a random (H, W) window.