pugh_torch.augmentations.transforms package

Submodules

pugh_torch.augmentations.transforms.resize_shortest module

class pugh_torch.augmentations.transforms.resize_shortest.ResizeShortest(length, interpolation=1, always_apply=False, p=1)

Bases: albumentations.core.transforms_interface.DualTransform

Resize the input so that the shortest side has the given pixel dimension.

Parameters

length (int) – Length in pixels of the resulting shortest side of the image.

apply(img, interpolation=1, **params)

apply_to_bbox(bbox, **params)

apply_to_keypoint(keypoint, **params)

Keypoints are not normalized and need their xy rescaled.

get_transform_init_args_names()

Module contents