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,可选,默认值 = ‘’) – 要添加到输出文件名的后缀。