利用Openvino開發(fā)AI應(yīng)用

主要分為如下幾步:

1. 創(chuàng)建Openvino的運(yùn)行時(shí)核心類

2. 為特定設(shè)備編譯AI模型

3. 創(chuàng)建推斷請(qǐng)求

4. 設(shè)置輸入

5. 開始推斷

6. 處理推斷結(jié)果

7. 構(gòu)建應(yīng)用和鏈接openvino運(yùn)行時(shí)庫

參考代碼結(jié)構(gòu):

#include <openvino/openvino.hpp>

ov::Core core;

ov::CompiledModel compiled_model = core.compile_model("model.onnx", "AUTO");

ov::InferRequest infer_request = compiled_model.create_infer_request();

// Get input port for model with one input

auto input_port = compiled_model.input();

// Create tensor from external memory

ov::Tensor input_tensor(input_port.get_element_type(), input_port.get_shape(), memory_ptr);

// Set input tensor for model with one input

infer_request.set_input_tensor(input_tensor);

infer_request.start_async();

infer_request.wait();

// Get output tensor by tensor name

auto output = infer_request.get_tensor("tensor_name");

const float *output_buffer = output.data<const float>();

// output_buffer[] - accessing output tensor data

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容