nvidia.dali.fn.experimental.peek_image_shape#

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

获取编码图像的形状。

此操作符返回图像解码后的形状。

注意

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

支持的后端

  • ‘cpu’

参数:

__input (TensorList) – 操作符的输入。

关键词参数:

  • adjust_orientation (bool, 可选, 默认 = True) – 计算形状时使用 EXIF 方向元数据。

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

    每个样本的输出大小提示（以字节为单位）。

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

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

  • image_type (nvidia.dali.types.DALIImageType, 可选, 默认 = DALIImageType.RGB) – 图像的颜色格式。

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

在本页中