昇腾社区首页
中文
注册
开发者
下载
支持

抠图

功能介绍

对输入的图像进行抠图操作，输出到Tensor对象中。

接口调用流程

使用抠图接口前，用户需准备好被抠图的图片并转换到Tensor对象中。

图1 张量方法（抠图）接口调用流程

关键步骤说明如下：

  1. 调用MxInit()接口进行全局初始化。
  2. 构造抠图Rect以及输出Tensor。

    根据业务要求，选择一对一或一对多的抠图方式，构造对应的输入的Rect和输出的Tensor。

  3. 选择执行方式进行抠图，请根据实际业务选择同步抠图方式或异步抠图方式。
    • 同步执行。
      • 不创建Stream，将输入图片及其他参数传入Crop方法，获取抠图结果。
    • 异步执行。
      1. 创建Stream，具体请参见异步调用。
      2. 将输入图片、已创建的Stream及其他参数传入Crop方法，获取抠图结果。
  4. 调用MxDeInit()接口对初始化的全局资源进行去初始化。

示例代码

以下为功能特性关键步骤的代码示例，不可以直接拷贝编译运行，仅供参考。 
//初始化
MxBase::MxInit();
{
    // 读取图片
    std::string imgPath = "./test.jpg";
    cv::Mat imgData = cv::imread(imgPath, cv::IMREAD_UNCHANGED);
    std::vector<uint32_t> shape{600, 600, 3};

    // 构造输入tensor
    MxBase::Tensor input(imgData.data, shape, MxBase::TensorDType::UINT8, -1);
    input.ToDvpp(0);

    // 设置抠图区域
    MxBase::Rect rect(0, 0, 320, 320);

    // 构造输出tensor
    std::vector<uint32_t> dstShape = {320, 320, 3};
    MxBase::Tensor dst(dstShape, MxBase::TensorDType::UINT8, -1);
    dst.Malloc();
    dst.ToDevice(0);

    // 设置输出tensor是否保留无效边界
    bool keepMargin = true;

    // 执行抠图操作
    APP_ERROR ret = MxBase::Crop(input, rect, dst, keepMargin);
    if (ret != APP_ERR_OK) {
        std::cout << "Crop failed." << std::endl;
    }
}
//去初始化
MxBase::MxDeInit();