nvidia.dali.fn.color_space_conversion#

nvidia.dali.fn.color_space_conversion(__input, /, *, bytes_per_sample_hint=[0], image_type, output_type, preserve=False, device=None, name=None)#

在各种图像颜色模型之间进行转换。

此操作符允许序列输入并支持体积数据。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

__input (TensorList ('FDHWC', 'FHWC', 'DHWC', 'HWC')) – 操作符的输入。

关键字参数:

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

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

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

  • image_type (nvidia.dali.types.DALIImageType) – 输入图像的色彩空间。

  • output_type (nvidia.dali.types.DALIImageType) – 输出图像的色彩空间。

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

另请参阅