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)[source]

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), always_apply=False, p=0.5)[source]

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).

  • 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)[source]

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)[source]

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)[source]

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)[source]

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)[source]

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=(0, 0, 1), always_apply=False, p=0.5)[source]
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)[source]

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