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 的 float) –

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

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

    支持 per-frame 输入。

  • axis (float 或 float 列表或 float 的 TensorList，可选，默认值 = []) –

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

    该向量不需要归一化，但必须具有非零长度。反转向量等效于更改 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 列表或 float 的 TensorList，可选，默认值 = []) –

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

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

另请参阅

在本页中