imgaug_transform

lightning_pose.data.augmentations.imgaug_transform(cfg: DictConfig) → Sequential

Create simple data transform pipeline that augments images.

Parameters:

cfg –

standard config file that carries around dataset info; relevant is the parameter “cfg.training.imgaug” which can take on the following values: - default: resizing only - dlc: imgaug pipeline implemented in DLC 2.0 package - dlc-top-down: dlc pipeline plus random flipping along both horizontal and vertical

axes

Returns:

imgaug pipeline