nvidia.dali.fn.multi_paste#

nvidia.dali.fn.multi_paste(__input_0, /, *__input_, bytes_per_sample_hint=[0], dtype=None, in_anchors=None, in_anchors_rel=None, in_ids=None, out_anchors=None, out_anchors_rel=None, output_size=None, preserve=False, shapes=None, shapes_rel=None, device=None, name=None)#

对图像批次执行多次粘贴到每个输出。

如果指定了 in_ids，则运算符正好需要一个输入批次。 在这种情况下，对于每个输出样本，in_ids 描述了应将输入批次中的哪些样本粘贴到输出批次中的相应样本。

如果省略 in_ids 参数，则运算符接受多个输入。 在这种情况下，来自每个输入批次的第 i 个样本将被粘贴到输出批次的第 i 个样本。 所有输入批次必须具有相同的类型和设备位置。

如果输入形状是统一的，并且没有提供显式的 output_size，则运算符假定相同的输出形状（输出画布大小）。 否则，必须指定 output_size。

此运算符还可以更改数据类型。

此运算符允许序列输入。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

  • __input_0 (TensorList ('HWC', 'FHWC')) – 运算符的输入。

  • __input_[1..63] (TensorList, 可选) – 此函数最多接受 63 个可选的位置输入

关键字参数:

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

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

    如果指定，则将预先分配位于 GPU 或分页锁定主机内存中的运算符输出，以适应此大小的样本批次。

  • dtype (nvidia.dali.types.DALIDataType, 可选) – 输出数据类型。 如果未设置，则使用输入类型。

  • in_anchors (int 或 TensorList of int, 可选) –

    源区域左上角的绝对坐标。

    锚点表示为 2D 张量，其中第一维度等于粘贴区域的数量，第二维度为 2（对于 H 和 W 范围）。

    如果既未提供 in_anchors 也未提供 in_anchors_rel，则所有锚点均为零。

    支持 per-frame 输入。

  • in_anchors_rel (float 或 TensorList of float, 可选) –

    源区域左上角的相对坐标。

    该参数的工作方式类似于 in_anchors，但值应为 [0, 1] 范围内的浮点数，描述相对于输入样本形状的锚点位置。

    支持 per-frame 输入。

  • in_ids (int 或 list of int 或 TensorList of int, 可选) –

    要从中粘贴数据的输入的索引。

    如果指定，则运算符正好接受一个批次作为输入。

  • out_anchors (int 或 TensorList of int, 可选) –

    输出画布中粘贴区域左上角的绝对坐标。

    锚点表示为 2D 张量，其中第一维度等于粘贴区域的数量，第二维度为 2（对于 H 和 W 范围）。

    如果既未提供 out_anchors 也未提供 out_anchors_rel，则所有锚点均为零，这使得所有粘贴区域都从输出画布的左上角开始。

    支持 per-frame 输入。

  • out_anchors_rel (float 或 TensorList of float, 可选) –

    输出画布中粘贴区域左上角的相对坐标。

    工作方式类似于 out_anchors 参数，但值应为 [0, 1] 范围内的浮点数，描述相对于输出画布大小的粘贴区域左上角。

    支持 per-frame 输入。

  • output_size (int 或 list of int 或 TensorList of int, 可选) –

    描述输出形状（即输出粘贴的画布大小）的元组 (H, W)。

    如果运算符以统一形状的输入运行，则可以省略。 在这种情况下，相同的形状用作画布大小。

  • preserve (bool, 可选, 默认 = False) – 阻止运算符从图中删除，即使其输出未使用。

  • shapes (int 或 TensorList of int, 可选) –

    粘贴区域的形状。

    形状表示为 2D 张量，其中第一维度等于粘贴区域的数量，第二维度为 2（对于 H 和 W 范围）。

    如果既未提供 shapes 也未提供 shapes_rel，则会计算形状，以使区域从输入锚点跨越到输入图像的末尾。

    支持 per-frame 输入。

  • shapes_rel (float 或 TensorList of float, 可选) –

    粘贴区域的相对形状。

    工作方式类似于 shape 参数，但值应为 [0, 1] 范围内的浮点数，描述相对于输入形状的粘贴区域形状。

    支持 per-frame 输入。

在此页面上