接口定义
HIAI_TENSOR_API_EXPORT std::shared_ptr<IImageTensorBuffer> CreateImageTensorBuffer(int32_t b, int32_t h, int32_t w, ImageFormat format, ImageColorSpace colorSpace, int32_t rotation);
功能介绍
DDK模型管家V2接口创建图像输入tensor。
参数
名称 | 类型 | 描述 |
b | int32_t | 输入图像的batch。 |
h | int32_t | 输入图像的高度。 |
w | int32_t | 输入图像的宽度。 |
format | ImageFormat | 输入图像的format。 |
colorSpace | ImageColorSpace | 输入图像的colorSpace。 |
rotation | int32_t | 输入图像的角度。 |
返回
类型 | 描述 |
std::shared_ptr<IImageTensorBuffer> | 输出参数。 |