Tensor类的属性列表
to_device
函数功能
移动Tensor到Device内存中。
函数原型
to_device(deviceId)
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| deviceId | int | Device设备的ID号。 |
抛异常接口
分配Tensor内存失败及拷贝到Device侧失败，抛出runtime异常。
父主题： Tensor类
to_host
函数功能
移动Tensor到Host内存中。
函数原型
to_host()
抛异常接口
分配Tensor内存失败及拷贝到Host侧失败，抛出runtime异常。
父主题： Tensor类
Tensor类的属性列表
| 属性名 | 说明 | 备注 |
|---|---|---|
| device | Tensor所在Device的ID | -1表示运行在Host侧，1表示设备号1，2表示设备号2，以此类推。 |
| dtype | Tensor数据类型 | base.dtype数据类型。 |
| shape | Tensor的维度信息 | 返回的是list数据类型。 |
父主题： Tensor类
Image
函数原型
Image()
Image(ndarray b, image_format format, tuple(Size, Size) imageSizeInfo = DEFAULT_IMAGE_SIZE_INFO)
函数功能
- 构造一个空的Image对象。
- 从Host侧的numpy的ndarray转为Image对象。
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| b | ndarray | 构造Image的numpy数组，各元素类型为np.uint8且形状为3维，依次表示图片的HWC（高、宽、通道数）。 |
| format | image_format | 图像色域格式，支持以下色域类型。 - base.yuv_400 - base.rgb - base.bgr - base.rgba - base.bgra - base.argb - base.abgr “b”（输入ndarray）形状中的通道数（C）需要与色域类型一致，各色域类型对应通道数参见如下。 - yuv_400为1。 - bgr、rgb为3。 - rgba、bgra、argb、abgr为4。 |
| imageSizeInfo | tuple | 图像有效宽高与对齐后宽高的组合，有效宽、高应不超过对齐后的宽、高，输入方式参见如下。 text<br>tuple(有效宽高，对齐后的宽高)<br>- 有效宽、高取值范围为[6, 8192]。 - 对齐后宽、高取值范围为[16, 8192]，其中宽为16的倍数、高为2的倍数。 - “b”（输入ndarray）形状中的高宽需要与有效宽高和对齐宽高其中之一保持一致。 - 默认值为“DEFAULT_IMAGE_SIZE_INFO”，其中有效宽高与对齐后宽高均为(0, 0)，若该参数为默认值，则自动根据“b”的形状获取有效宽高和对齐宽高。 |
:::note 说明
- 从opencv读取图片转换的ndarray，若形状为WHC（宽，高，通道数），需要进行转置为HWC（高，宽，通道数）之后进行构造。
- 从get_tensor/to_tensor/get_orginal_tensor接口获取的tensor，转换为ndarray后，形状为NHWC（个数，高，宽，通道数）或NHW（个数，高，宽），需要根据实际情况截取或扩展到HWC之后进行构造。 :::
输出参数说明
Image对象。
父主题： Image类
to_tensor
函数功能
将Image对象转换为Tensor类对象。
函数原型
to_tensor()
输出参数说明
返回Tensor。
父主题： Image类
在线提单