nvidia.dali.fn.peek_image_shape#

nvidia.dali.fn.peek_image_shape(__input, /, *, bytes_per_sample_hint=[0], dtype=DALIDataType.INT64, preserve=False, device=None, name=None)#

获取编码图像的形状。

此算子返回图像解码后的形状。

注意

不建议将此算子与动态执行器（pipeline 构造函数中的 exec_dynamic=True）一起使用。请在解码后的图像上改用 nvidia.dali.pipeline.DataNode.shape()。

支持的后端

  • ‘cpu’

参数：:

__input (TensorList) – 算子的输入。

关键字参数：:

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

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

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

  • dtype (nvidia.dali.types.DALIDataType, 可选, 默认 = DALIDataType.INT64) – 数据类型，尺寸将转换为该类型。

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

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

    警告

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

在本页