nvidia.dali.fn.rotate#

nvidia.dali.fn.rotate(__input, /, *, angle, axis=[], bytes_per_sample_hint=[0], dtype=None, fill_value=None, interp_type=DALIInterpType.INTERP_LINEAR, keep_size=False, preserve=False, size=[], device=None, name=None)#

按指定角度旋转图像。

此操作器支持体积数据。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

__input (TensorList ('HWC', 'FHWC', 'DHWC', 'FDHWC')) – 操作器的输入。

关键词参数:

  • angle (float 或 TensorList of float) –

    图像旋转的角度，以度为单位。

    对于二维数据，假设左上角位于 (0,0)，则旋转是逆时针的。对于三维数据，angle 是绕提供的轴的正向旋转。

    支持 per-frame 输入。

  • axis (float 或 float 列表 或 TensorList of float, 可选, 默认 = []) –

    仅适用于三维，并且是图像绕其旋转的轴。

    该向量不需要归一化，但必须具有非零长度。反转向量等效于更改 angle 的符号。

    支持 per-frame 输入。

  • bytes_per_sample_hint (int 或 int 列表, 可选, 默认 = [0]) –

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

    如果指定，则将预先分配驻留在 GPU 或页锁定主机内存中的操作器输出，以容纳此大小的样本批次。

  • dtype (nvidia.dali.types.DALIDataType, 可选) –

    输出数据类型。

    如果未设置，则使用输入类型。

  • fill_value (float, 可选) –

    用于填充源图像外部区域的值。

    如果未指定值，则源坐标将被钳制，并且边框像素将被重复。

  • interp_type (nvidia.dali.types.DALIInterpType, 可选, 默认 = DALIInterpType.INTERP_LINEAR) – 使用的插值类型。

  • keep_size (bool, 可选, 默认 = False) –

    如果为 True，则保留原始画布大小。

    如果设置为 False（默认值），并且未设置大小，则会调整画布大小以容纳旋转后的图像，并尽可能减少填充。

  • output_dtype (nvidia.dali.types.DALIDataType) –

    警告

    参数 output_dtype 是 dtype 的已弃用别名。请改用 dtype。

  • preserve (bool, 可选, 默认 = False) – 即使操作器的输出未使用，也阻止将其从图中删除。

  • size (float 或 float 列表 或 TensorList of float, 可选, 默认 = []) –

    输出大小，以像素/点为单位。

    非整数大小将四舍五入为最接近的整数。应排除通道维度（例如，对于 RGB 图像，请指定 (480,640)，而不是 (480,640,3)）。

另请参阅

在本页中