acldvppCropAndResize
Atlas 200/300/500 推理产品不支持该算子。
Atlas 训练系列产品不支持该算子。
Atlas 推理系列产品不支持该算子。
Atlas 200/500 A2推理产品不支持该算子。
函数功能
算子功能:对图像做抠图缩放。
函数原型
每个算子有两段接口,必须先调用“acldvppCropAndResizeGetWorkspaceSize”接口获取入参并根据计算流程计算所需workspace大小,再调用“acldvppCropAndResize”接口执行计算。两段式接口如下:
- acldvppStatus acldvppCropAndResizeGetWorkspaceSize(const aclTensor *self, uint32_t top, uint32_t left, uint32_t height, uint32_t width, const aclIntArray* size, uint32_t interpolationMode, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
- acldvppStatus acldvppCropAndResize(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
acldvppCropAndResizeGetWorkspaceSize
- 接口定义:
acldvppStatus acldvppCropAndResizeGetWorkspaceSize(const aclTensor *self, uint32_t top, uint32_t left, uint32_t height, uint32_t width, const aclIntArray* size, uint32_t interpolationMode, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
- 参数说明:
- self:表示算子输入Tensor,需调用aclCreateTensor接口创建aclTensor类型的数据(数据存放在Device侧),输入Tensor的dataType支持UINT8/FLOAT、Format支持NCHW/NHWC、不支持非连续的Tensor,同时N支持1或空、C支持1或3(1表示输入GRAY图,3表示输入RGB图)。
- top:抠图的上边界位置。
- left:抠图的左边界位置。
- height:抠图的高度,取值范围[4, 32768]。
- width:抠图的宽度,取值范围[6, 32768]。
- size:表示缩放之后的高、宽。需调用aclCreateIntArray接口创建参数aclIntArray类型的数据,数组长度为2,第一个元素是高、第二个元素是宽。缩放后宽高必须与输出宽高一致。
- interpolationMode:表示缩放算法,该参数的取值范围如下:
- 0:bilinear算法
- 1:nearest算法
- 2:bicubic算法
- out:表示算子输出Tensor,需调用aclCreateTensor接口创建aclTensor类型的数据(数据存放在Device侧),输出Tensor的dataType支持UINT8/FLOAT、Format支持NCHW/NHWC、不支持非连续的Tensor,同时N只支持1、C支持1或3(1表示GRAY图,3表示RGB图),dataType、Format需要和self一致,Shape中的N轴、C轴大小需要和self一致,Shape中的W轴、H轴大小需和缩放后的宽、高保持一致。
- workspaceSize:返回用户需要在Device侧申请的workspace大小。
- executor:返回op执行器,包含了算子计算流程。
- 返回值:
返回acldvppStatus状态码,具体请参见acldvpp返回码。
acldvppCropAndResize
- 接口定义:
acldvppStatus acldvppCropAndResize(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
- 参数说明:
- workspace:需调用aclrtMalloc接口申请Device内存,内存大小为workspaceSize,aclrtMalloc接口输出的内存地址在此处传入。
- workspaceSize:与acldvppCropAndResizeGetWorkspaceSize接口获取的workspaceSize保持一致。
- executor:op执行器,包含了算子计算流程,与acldvppCropAndResizeGetWorkspaceSize接口的executor保持一致。
- stream:指定执行任务的Stream,可复用已创建的Stream节省资源或调用aclrtCreateStream接口创建Stream,再作为入参在此处传入。
- 返回值:
返回acldvppStatus状态码,具体请参见acldvpp返回码。
约束与限制
- 支持图像分辨率范围在[6*4~32768*32768]。
- 当输出out的shape的宽度超过4096时,输入self的shape的宽高不能小于128*16。
父主题: DVPP算子接口