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

acldvppErase

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

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

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

Atlas 200/500 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。

搜索结果
找到“0”个结果

当前产品无相关内容

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