tensor_to_image

函数功能

将Host侧Tensor类转换为Host侧Image类,或将Device侧Tensor类转换为DVPP侧Image类。

当前接口仅能够在Atlas 推理系列产品Atlas 800I A2推理产品环境上调用。

在转换过程中,会对Image的宽进行16向上对齐、对Image的高进行2向上对齐,转换后的Image类对象将存在补边区域。

转换后的Image类可通过类对象属性width和height查看补边后的宽、高,通过类对象属性original_width和original_height查看图片原始宽、高。

例如:

函数原型

1
tensor_to_image(inputTensor: Tensor, imageFormat: image_format) 

参数说明

参数名

类型

说明

inputTensor

Tensor类

Tensor类,输入张量。输入需满足以下要求。

  • 元素类型需为Uint8类型。
  • Tensor的维度需为2(YUV400格式时)、3、4。
  • Tensor的宽、高、通道数需与image_format相匹配。

imageFormat

image_format枚举类

指定图片的格式,需与inputTensor数据所对应的图片格式相匹配。

返回参数说明

返回对应的Image对象。