acldvppEncodeJpeg

支持的产品型号

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

函数功能

算子功能：将单通道（GRAY）或三通道（RGB）图像编码为JPEG图像。

函数原型

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

• acldvppStatus acldvppEncodeJpegGetWorkspaceSize(const aclTensor* self, uint32_t quality, aclTensor* out, uint64_t* workspaceSize, aclOpExecutor** executor)
• acldvppEncodeJpeg(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, aclrtStream stream)

acldvppEncodeJpegGetWorkspaceSize

• 接口定义：

  acldvppStatus acldvppEncodeJpegGetWorkspaceSize(const aclTensor* self, uint32_t quality, aclTensor* out, uint64_t* workspaceSize, aclOpExecutor** executor)

• 参数说明：
  • self：表示算子输入Tensor，需调用aclCreateTensor接口创建aclTensor类型的数据（数据存放在Device侧），输入Tensor的dataType支持UINT8、Format支持NCHW/NHWC、不支持非连续的Tensor，同时N支持1或空、C支持1或3（1表示输入GRAY图，3表示输入RGB图）
  • quality：指定编码质量，范围[1, 100]，数值越小图片质量越差，但图片数据量小，占用内存少；数值越大图像质量越高，但图片数据量大，占用内存多。
  • out：表示算子输出Tensor，需调用aclCreateTensor接口创建aclTensor类型的数据（数据存放在Device侧），输出Tensor的dataType支持UINT8、Format支持ND、不支持非连续的Tensor。
  • workspaceSize：返回用户需要在Device侧申请的workspace大小。
  • executor：返回op执行器，包含了算子计算流程。
• 返回值：

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

acldvppEncodeJpeg

• 接口定义：

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

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

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

约束与限制

支持分辨率范围在[32*32, 8192*8192]内的JPEG图像编码。

原始图像的宽高需要是2对齐。