torch_geometric.transforms.NormalizeRotation

class NormalizeRotation(max_points: int = -1, sort: bool = False)[source]

Bases: BaseTransform

Rotates all points according to the eigenvectors of the point cloud (functional name: normalize_rotation). If the data additionally holds normals saved in data.normal, these will be rotated accordingly.

Parameters:
  • max_points (int, optional) – If set to a value greater than 0, only a random number of max_points points are sampled and used to compute eigenvectors. (default: -1)

  • sort (bool, optional) – If set to True, will sort eigenvectors according to their eigenvalues. (default: False)