s6.nn.data.aug

Ordered augmentation helpers for image/keypoint/mask samples.

class s6.nn.data.aug.KeypointAugmentationSample(image: ndarray, keypoint: ndarray, mask: ndarray | None = None)

Bases: object

Image/keypoint/mask payload passed between ordered augmentation stages.

image: ndarray

keypoint: ndarray

mask: ndarray | None = None

class s6.nn.data.aug.KeypointAugmentationStage(name: str, apply: Callable[[KeypointAugmentationSample], KeypointAugmentationSample])

Bases: object

One named augmentation stage in a sequential augmentation pipeline.

name: str

apply: Callable[[KeypointAugmentationSample], KeypointAugmentationSample]

class s6.nn.data.aug.OrderedKeypointAugmentation(stages: Sequence[KeypointAugmentationStage])

Bases: object

Apply keypoint augmentation stages in the order they are configured.

property stage_order: Tuple[str, ...]

Configured execution order.