Augmentation pipeline block

[talmo]

Block/module for flexible augmentation. This should be performant and flexible with respect to the specific augmentations so we can add more in the future.

• Implement backend(s)
  • Explore using Kornia if possible to accelerate operations on GPU
  • Also support Albumentations (CPU-only?)
• Make sure (and add tests) for ensuring that augmentations are applied to both the image and the keypoints!
• Implement core augmentations
  • Rotations
  • Scaling
• Implement additional augmentations
  • Contrast/brightness
  • Cropping
  • Gaussian blur
  • Dropout patches
  • MixUp

Notes:

• Topdown pipelines that operate on instance-level examples may benefit from specialized augmentation (see Top-down centered-instance pipeline #3 notes)
• Implement frame-level augmentation to start for generality
• If possible, try to keep it general enough to handle both single frame and multi-frame by operating on batches (where single-frame is special case of batch size = 1). This will make it easier to generalize to multi-frame pipelines later.