c_cpp Atlas300代码片段

Posted 2021-05-14

    ImageData<uint8_t> resized_image;
    int out_size = ALIGN_UP(input_tensor_dim_vec_[model_id].w, 128) * ALIGN_UP(input_tensor_dim_vec_[model_id].h, 16) * 3/2;
    uint8_t * out_buffer = (uint8_t *)memalign(128, out_size);
    if (out_buffer == NULL)
    {
        HIAI_ENGINE_LOG("Hfbc2Yuv call memalign failed");
        return HIAI_ERROR;
    }
    resized_image.size = out_size;
    resized_image.width = input_tensor_dim_vec_[model_id].w;
    resized_image.height = input_tensor_dim_vec_[model_id].h;
    resized_image.data.reset(out_buffer, std::default_delete<uint8_t []>());