acldvppRotate

支持的产品型号

Atlas A2训练系列产品/Atlas 800I A2推理产品

函数功能

算子功能：对图像做逆时针方向的旋转。旋转示意图如下：

函数原型

每个算子有两段接口，必须先调用“acldvppRotateGetWorkspaceSize”接口获取入参并根据计算流程计算所需workspace大小，再调用“acldvppRotate”接口执行计算。两段式接口如下：

acldvppStatus acldvppRotateGetWorkspaceSize(const aclTensor *self, float angle, uint32_t interpolationMode, bool expand, const aclIntArray *center, uint32_t paddingMode, const aclFloatArray *fill, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
acldvppStatus acldvppRotate(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

acldvppRotateWorkspaceSize

接口定义：
acldvppStatus acldvppRotateGetWorkspaceSize(const aclTensor *self, float angle, uint32_t interpolationMode, bool expand, const aclIntArray *center, uint32_t paddingMode, const aclFloatArray *fill, 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图）。
- angle: 旋转的角度，以度为单位，逆时针方向。
- expand: 是否扩展图像，取值说明如下：
  - True：扩展图像尺寸大小使其足以容纳整个旋转图像。请注意，扩展时将假设图像为中心旋转且未进行平移。
    expand为True时，输出宽高计算方法如下：
    
    将图像四个顶点的坐标代入如下旋转矩阵公式中，计算出水平和垂直方向的最大距离，分别作为输出图像的宽高
    
    $\text{[math]}$
  - False：保持图像尺寸大小不变。
- center: 旋转中心。
  当expand为False时，如果center设置为nullptr，表示按照图像中心点旋转；如果设置center，则按照center设置的中点心旋转，以图像左上角为原点设置center的坐标值。
  
  当expand为True时，设置center不生效，固定按照图像中心点旋转。
- interpolationMode：表示插值算法，该参数的取值范围如下：
  - 0：bilinear算法
  - 1：nearest算法
- paddingMode：填充模式，该参数取值如下：
  - 0：CONSTANT，填充固定值
  - 1：EDGE，重复最后一个元素。举例，其中*表示任意图像元素: aaaaaa|a*****h|hhhhhhh
- fill: fill是一个长度为3的数组，用于设置每个通道上填充的值，需调用aclCreateFloatArray接口创建参数aclFloatArray类型的数据，仅在paddingMode为CONSTANT时fill参数有效。如果C为1（表示GRAY图），填充fill[0]；如果C为3（表示RGB图），按照R、G、B顺序依次填写fill[0]、fill[1]、fill[2]。
- 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一致。
- workspaceSize：返回用户需要在Device侧申请的workspace大小。
- executor：返回op执行器，包含了算子计算流程。
返回值：
返回acldvppStatus状态码，具体请参见acldvpp返回码。

acldvppRotate

接口定义：
acldvppStatus acldvppRotate(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, aclrtStream stream);
参数说明：
- workspace：需调用aclrtMalloc接口申请Device内存，内存大小为workspaceSize，aclrtMalloc接口输出的内存地址在此处传入。
- workspaceSize：与acldvppRotateGetWorkspaceSize接口获取的workspaceSize保持一致。
- executor：op执行器，包含了算子计算流程，与acldvppRotateGetWorkspaceSize接口的executor保持一致。
- stream：指定执行任务的Stream，可复用已创建的Stream节省资源或调用aclrtCreateStream接口创建Stream，再作为入参在此处传入。
返回值：
返回acldvppStatus状态码，具体请参见acldvpp返回码。

约束与限制

支持图像分辨率范围在[6*4~32768*32768]。
输入输出格式保持一致。

父主题： DVPP算子接口