nvidia.dali.fn.transforms.rotation#

nvidia.dali.fn.transforms.rotation(__input=None, /, *, angle, axis=None, bytes_per_sample_hint=[0], center=None, preserve=False, reverse_order=False, device=None, name=None)#

生成旋转仿射变换矩阵。

如果将另一个变换矩阵作为输入传递，则此算子将对提供的矩阵应用旋转。

如果提供了旋转轴，则维度数假定为 3，否则为 2。

注意

此算子的输出可以直接馈送到 CoordTransform 和 WarpAffine 算子。

此算子允许序列输入。

支持的后端

  • ‘cpu’

参数:

__input (TensorList, optional) – 算子的输入。

关键字参数:

  • angle (float 或 float 的 TensorList) –

    角度，以度为单位。

    支持 per-frame 输入。

  • axis (float 或 float 的列表或 float 的 TensorList, optional) –

    旋转轴（仅适用于 3D 变换）。

    向量不需要归一化，但必须具有非零长度。

    反转向量等效于更改 angle 的符号。

    支持 per-frame 输入。

  • bytes_per_sample_hint (int 或 int 列表, optional, default = [0]) –

    输出大小提示，以每样本字节数为单位。

    如果指定，则算子的输出（位于 GPU 或分页锁定主机内存中）将被预先分配以容纳此大小的一批样本。

  • center (float 或 float 列表或 float 的 TensorList, optional) –

    旋转中心。

    如果提供，则元素数量应与变换的维度匹配。

    支持 per-frame 输入。

  • preserve (bool, optional, default = False) – 阻止算子从图中移除，即使其输出未使用。

  • reverse_order (bool, optional, default = False) –

    确定组合仿射变换时的顺序。

    如果设置为 False（默认），则算子的仿射变换将应用于输入变换。如果设置为 True，则输入变换将应用于算子的变换。

    如果没有输入，则忽略此参数。

在本页中