下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

acldvppRotate

Atlas 200/300/500 推理产品不支持该算子。

Atlas 训练系列产品不支持该算子。

Atlas 推理系列产品不支持该算子。

Atlas 200/500 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时,输出宽高计算方法如下:

        将图像四个顶点的坐标代入如下旋转矩阵公式中,计算出水平和垂直方向的最大距离,分别作为输出图像的宽高

      • 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]。
  • 输入输出格式保持一致。
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词