下载
中文
注册

缩放

功能介绍

对输入的图像进行缩放操作,输出到Tensor对象中。

接口调用流程

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

图1 张量方法(缩放)接口调用流程

关键步骤说明如下:

  1. 调用MxInit()接口进行全局初始化。
  2. 构造缩放参数和输出Tensor。

    根据业务要求,构造对应的输入Size和Tensor以及输出Tensor。

  3. 选择执行方式进行缩放,请根据实际业务选择同步缩放方式或异步缩放方式。
    1. 同步执行。
      1. 不创建Stream,将输入图片及其他参数传入Resize方法,获取缩放结果。
    2. 异步执行。
      1. 创建Stream,具体请参见异步调用
      2. 将输入图片、已创建的Stream及其他参数传入Resize方法,获取缩放结果。
  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::Size resize(289, 289);

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

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

    // 执行缩放操作
    APP_ERROR ret = MxBase::Resize(input, dst, resize, MxBase::Interpolation::BILINEAR_SIMILAR_OPENCV, keepMargin);
    if (ret != APP_ERR_OK) {
        std::cout << "Resize failed." << std::endl;
    }
}

// 去初始化
MxBase::MxDeInit();