get_imgaug_transform

lightning_pose.utils.scripts.get_imgaug_transform(cfg: DictConfig) Sequential[source]

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