ConvertToTensor
函数功能
将DVPP侧Image类转换为Device侧Tensor类,或将Host侧Image转换为Host侧Tensor,转换后的Tensor类对象的数据类型为uint8。
- 若转换失败,则会返回空的Tensor实例。
- 若因内存不足或无法识别芯片等异常场景,导致Tensor类构造失败的话,则会抛出异常。
 
 - “withStride”参数为“true”时,Tensor对象会保留Image对象的补边信息。为减少内存拷贝,提高运行效率,此时返回的Tensor对象与Image对象共享数据内存,同时在该场景下,Image对象的数据内存会与Tensor对象的数据内存相互影响,例如:在Image对象释放后,对应的Tensor对象中的数据将变为无效内存数据。
- ConvertToTensor()为无参数接口时,返回的Tensor对象会保留Image对象的补边信息并存在batch维度(即NHWC,其中batch维度N = 1)。
函数原型
Tensor Image::ConvertToTensor(bool withStride, bool formatNHWC); Tensor Image::ConvertToTensor();
参数说明
| 参数名 | 输入/输出 | 说明 | 
|---|---|---|
| withStride | 输入 | bool类型,指定是否保留补边信息。 
 | 
| formatNHWC | 输入 | bool类型,指定转换得到的Tensor是否有batch维度。 
 | 
返回参数说明
| 数据结构 | 说明 | 
|---|---|
| Tensor | Tensor类,具体请参见Tensor。 | 
父主题: Image