Augmentations (volumentations.augmentations)

Transforms

class volumentations.augmentations.transforms.Scale3d(scale_limit=(0.1, 0.1, 0.1), bias=(1, 1, 1), always_apply=False, p=0.5)

Scale the input point cloud.

Parameters

• scale_limit (float, float, float) – maximum scaling of input point cloud. Default: (0.1, 0.1, 0.1).

• bias (list(float, float, float)) – base scaling that is always applied. List of 3 values to determine the basic scaling. Default: (1, 1, 1).

• p (float) – probability of applying the transform. Default: 0.5.

Targets:

points normals features labels

class volumentations.augmentations.transforms.RotateAroundAxis3d(rotation_limit=1.5707963267948966, axis=(0, 0, 1), center_point=None, always_apply=False, p=0.5)

Rotate point cloud around axis on random angle.

Parameters

• rotation_limit (float) – maximum rotation of the input point cloud. Default: (pi / 2).

• axis (list(float, float, float)) – axis around which the point cloud is rotated. Default: (0, 0, 1).

• center_point (float, float, float) – point around which to rotate. Default: mean points.

• p (float) – probability of applying the transform. Default: 0.5.

Targets:

points normals features labels

class volumentations.augmentations.transforms.Crop3d(x_min=-inf, y_min=-inf, z_min=-inf, x_max=inf, y_max=inf, z_max=inf, always_apply=False, p=1.0)

Crop region from image.

Parameters

• x_min (float) – Minimum x coordinate.

• y_min (float) – Minimum y coordinate.

• z_min (float) – Minimum z coordinate.

• x_max (float) – Maximum x coordinate.

• y_max (float) – Maximum y coordinate.

• z_max (float) – Maximum z coordinate.

• p (float) – probability of applying the transform. Default: 0.5.

Targets:

points normals features labels

class volumentations.augmentations.transforms.RandomMove3d(x_min=-1.0, y_min=-1.0, z_min=-1.0, x_max=1.0, y_max=1.0, z_max=1.0, offset=(0, 0, 0), always_apply=False, p=0.5)

Move point cloud on random offset.

Parameters

• x_min (float) – Minimum x coordinate. Default: -1.

• y_min (float) – Minimum y coordinate. Default: -1.

• z_min (float) – Minimum z coordinate. Default: -1.

• x_max (float) – Maximum x coordinate. Default: 1.

• y_max (float) – Maximum y coordinate. Default: 1.

• z_max (float) – Maximum z coordinate. Default: 1.

• p (float) – probability of applying the transform. Default: 0.5.

Targets:

points normals features labels

class volumentations.augmentations.transforms.Move3d(offset=(0, 0, 0), always_apply=False, p=1.0)

Move point cloud on offset.

Parameters

• offset (float) – coorinate where to move origin of coordinate frame. Default: 0.

• p (float) – probability of applying the transform. Default: 0.5.

Targets:

points normals features labels

class volumentations.augmentations.transforms.Center3d(offset=(0, 0, 0), always_apply=False, p=0.5)

Move average of point cloud and move it to coordinate (0,0,0).

Parameters

p (float) – probability of applying the transform. Default: 0.5.

Targets:

points normals features labels

class volumentations.augmentations.transforms.RandomDropout3d(dropout_ratio=0.2, always_apply=False, p=0.5)

Randomly drop points from point cloud.

Parameters

• dropout_ratio (float) – Percent of points to drop. Default: 0.2.

• p (float) – probability of applying the transform. Default: 0.5.

Targets:

points normals features labels

class volumentations.augmentations.transforms.Flip3d(axis=(1, 0, 0), always_apply=False, p=0.5)

Flip point cloud around axis

Implemented as rotation on 180 deg around axis.

Parameters

• axis (list(float, float, float)) – Axis to flip the point cloud around. Default: 0.2.

• p (float) – probability of applying the transform. Default: 0.5.

Targets:

points normals features labels

Functionals

volumentations.augmentations.functional.rotate_around_axis(points, axis, angle, center_point=None)

Return the rotation matrix associated with counterclockwise rotation about the given axis by angle in radians. https://stackoverflow.com/questions/6802577/rotation-of-3d-vector