文档
注册
评分
提单
论坛
小AI

acldvppConvertColor

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

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

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

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

函数功能

算子功能:更改图像的色彩空间。

函数原型

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

  • acldvppStatus acldvppConvertColorGetWorkspaceSize(const aclTensor *self, acldvppConvertMode convertMode, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
  • acldvppStatus acldvppConvertColor(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

acldvppConvertColorGetWorkspaceSize

  • 接口定义:

    acldvppStatus acldvppConvertColorGetWorkspaceSize(const aclTensor *self, acldvppConvertMode convertMode, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)

  • 参数说明:
    • self:表示算子输入Tensor,需调用aclCreateTensor接口创建aclTensor类型的数据(数据存放在Device侧),输入Tensor的dataType支持UINT8/FLOAT、Format支持NHWC、不支持非连续的Tensor,同时N支持1或空、C支持1或3或4(1表示输入GRAY图,3表示输入RGB图,4表示包含alpha通道)。
    • convertMode:图像色彩空间转换的模式(枚举值与OpenCV保持一致)。若设置的转换模式涉及alpha通道,则只支持UINT8类型输入和输出数据。convertMode参数的取值范围如下:
      • COLOR_BGR2BGRA: 将BGR图像转换为BGRA图像。
      • COLOR_RGB2RGBA: 将RGB图像转换为RGBA图像。
      • COLOR_BGRA2BGR: 将BGRA图像转换为BGR图像。
      • COLOR_RGBA2RGB: 将RGBA图像转换为RGB图像。
      • COLOR_BGR2RGBA: 将BGR图像转换为RGBA图像。
      • COLOR_RGB2BGRA: 将RGB图像转换为BGRA图像。
      • COLOR_RGBA2BGR: 将RGBA图像转换为BGR图像。
      • COLOR_BGRA2RGB: 将BGRA图像转换为RGB图像。
      • COLOR_BGR2RGB: 将BGR图像转换为RGB图像。
      • COLOR_RGB2BGR: 将RGB图像转换为BGR图像。
      • COLOR_BGRA2RGBA: 将BGRA图像转换为RGBA图像。
      • COLOR_RGBA2BGRA: 将RGBA图像转换为BGRA图像。
      • COLOR_BGR2GRAY: 将BGR图像转换为GRAY图像。
      • COLOR_RGB2GRAY: 将RGB图像转换为GRAY图像。
      • COLOR_GRAY2BGR: 将GRAY图像转换为BGR图像。
      • COLOR_GRAY2RGB: 将GRAY图像转换为RGB图像。
      • COLOR_GRAY2BGRA: 将GRAY图像转换为BGRA图像。
      • COLOR_GRAY2RGBA: 将GRAY图像转换为RGBA图像。
      • COLOR_BGRA2GRAY: 将BGRA图像转换为GRAY图像。
      • COLOR_RGBA2GRAY: 将RGBA图像转换为GRAY图像。
    • out:表示算子输出Tensor,需调用aclCreateTensor接口创建aclTensor类型的数据(数据存放在Device侧),输出Tensor的dataType支持UINT8/FLOAT、Format支持NHWC、不支持非连续的Tensor,同时N支持1或空、C支持1或3或4(1表示输入GRAY图,3表示输入RGB图,4表示包含alpha通道),dataType、Format需要和self一致。
    • workspaceSize:返回用户需要在Device侧申请的workspace大小。
    • executor:返回op执行器,包含了算子计算流程。
  • 返回值:

    返回acldvppStatus状态码,具体请参见acldvpp返回码

acldvppConvertColor

  • 接口定义:

    acldvppStatus acldvppConvertColor(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

  • 参数说明:
    • workspace:需调用aclrtMalloc接口申请Device内存,内存大小为workspaceSize,aclrtMalloc接口输出的内存地址在此处传入。
    • workspaceSize:与acldvppConvertColorGetWorkspaceSize接口获取的workspaceSize保持一致。
    • executor:op执行器,包含了算子计算流程,与acldvppConvertColorGetWorkspaceSize接口的executor保持一致。
    • stream:指定执行任务的Stream,可复用已创建的Stream节省资源或调用aclrtCreateStream接口创建Stream,再作为入参在此处传入。
  • 返回值:

    返回acldvppStatus状态码,具体请参见acldvpp返回码

约束与限制

  • 支持图像分辨率范围在[6*4~4096*8192]。
  • 涉及alpha通道的转换模式,只支持UINT8类型输入和输出数据。
搜索结果
找到“0”个结果

当前产品无相关内容

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