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 输入。

在本页中