acldvppErase

支持的产品型号

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

函数功能

算子功能：使用指定的值擦除输入图像中指定范围内的像素点。

函数原型

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

• acldvppStatus acldvppEraseGetWorkspaceSize(const aclTensor *self, uint32_t top, uint32_t left, uint32_t height, uint32_t width, const aclFloatArray* value, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
• acldvppStatus acldvppErase(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

acldvppEraseGetWorkspaceSize

• 接口定义：

  acldvppStatus acldvppEraseGetWorkspaceSize(const aclTensor *self, uint32_t top, uint32_t left, uint32_t height, uint32_t width, const aclFloatArray* value, 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图）。

    当Tensor的dataType为FLOAT时，期望其值的范围为[0, 1]。

  • top：擦除区域左上角位置的纵坐标。
  • left：擦除区域左上角位置的横坐标。
  • height：擦除区域的高度。
  • width：擦除区域的宽度。
  • value： 表示在擦除区域的像素填充值。需调用aclCreateFloatArray接口创建aclFloatArray类型的数据，长度与输入Tensor的C通道数保持一致，当长度为3时，value分别用于填充R、G、B通道。

    如果传入空指针，则value值默认为0。

    当输入Tensor的dataType为FLOAT，value值范围为[0, 1]；当输入Tensor的dataType为UINT8，value值范围为[0，255]。

  • out：表示算子输出Tensor，需调用aclCreateTensor接口创建aclTensor类型的数据（数据存放在Device侧），输出Tensor的dataType支持UINT8/FLOAT、Format支持NCHW/NHWC、不支持非连续的Tensor，同时N只支持1或空、C支持1或3（1表示GRAY图，3表示RGB图），dataType、Format、Shape需要和self一致。
  • workspaceSize：返回用户需要在Device侧申请的workspace大小。
  • executor：返回op执行器，包含了算子计算流程。
• 返回值：

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

acldvppErase

• 接口定义：

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

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

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

约束与限制

支持图像分辨率范围在[6*4~4096*8192]。

dataType为UINT8时，结果与Torchvision一致；dataType为FLOAT时，erase区域结果与Torchvision最大差异为+-1/255。