torch_geometric.transforms.NormalizeRotation

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

Bases: BaseTransform

根据点云的特征向量旋转所有点 (函数名称:normalize_rotation)。 如果数据还包含保存在data.normal中的法线,这些法线也将相应旋转。

Parameters:
  • max_points (int, 可选) – 如果设置为大于 0 的值, 则仅随机采样 max_points 个点并用于计算特征向量。(默认值:-1

  • sort (bool, optional) – 如果设置为 True,将根据特征值对特征向量进行排序。(默认值:False