nvidia.dali.fn.dump_image

nvidia.dali.fn.dump_image(__input, /, *, bytes_per_sample_hint=[0], input_layout='HWC', preserve=False, suffix='', device=None, name=None)

将批量图像以 PPM 格式保存到磁盘。

对调试很有用。

支持的后端

• ‘cpu’

• ‘gpu’

参数:

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

关键词参数:

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

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

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

• input_layout (layout str, 可选, 默认 = ‘HWC’) – 输入图像的布局。

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

• suffix (str, 可选, 默认 = ‘’) – 要添加到输出文件名的后缀。