acldvppConvertColor
支持的产品型号
Atlas A2训练系列产品/Atlas 800I 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类型输入和输出数据。
父主题： DVPP算子接口
