【计算机视觉】使用opencv中dnn模块及openvino2022进行模型推理(C++接口)
admin
2024-03-01 06:04:58
0次

目录

  • 使用openvino:(推理使用由onnx生成的xml及bin文件)
      • 0.查看支持的推理设备:
      • 1.构造模型:
      • 2.进行预处理工作,cv::mat转ov::tensor
      • 3.推理后结果提取:
  • 使用opencv自带的dnn模块进行推理
      • 0.模型构造
      • 1.数据准备:原始mat变形为cv::dnn::Net的输入mat
      • 2.推理后结果提取:

使用openvino:(推理使用由onnx生成的xml及bin文件)

0.查看支持的推理设备:

ov::Core ie;
vector availableDevices = ie.get_available_devices();
for (int i = 0; i < availableDevices.size(); i++) {qDebug()<<"supported device name : "<

1.构造模型:

ov::Core ie;
auto network = ie.read_model(xml,bin);
if(gpu)
{
auto compiled_model = ie.compile_model(network, "GPU");
infer_request = compiled_model.create_infer_request();
}
else
{
auto compiled_model = ie.compile_model(network, "CPU");
infer_request = compiled_model.create_infer_request();
}

2.进行预处理工作,cv::mat转ov::tensor

计算机内存float*是一个一维数组线性存储,故相互赋值需将mat及tensor的索引统一为线性索引:从头逐一连续赋值;
一般而言有:

void cvImageToTensor(const cv::Mat & image, float *tensor, nvinfer1::Dims dimensions)
{const size_t channels = dimensions.d[1];const size_t height = dimensions.d[2];const size_t width = dimensions.d[3];// TODO: validate dimensions matchconst size_t stridesCv[3] = { width * channels, channels, 1 };const size_t strides[3] = { height * width, width, 1 };for (int i = 0; i < height; i++) {for (int j = 0; j < width; j++) {for (int k = 0; k < channels; k++) {const size_t offsetCv = i * stridesCv[0] + j * stridesCv[1] + k * stridesCv[2];const size_t offset = k * strides[0] + i * strides[1] + j * strides[2];tensor[offset] = (float) image.data[offsetCv];}}}
}

或使用opencv自己的取值函数at<>:

cv::Mat img = this->resize_image(frame, &newh, &neww, &padh, &padw);ov::Tensor input_tensor1 = infer_request.get_input_tensor(0);auto data1 = input_tensor1.data();cv::cvtColor(img,img,cv::COLOR_BGR2RGB);for (int h = 0; h < 640; h++){for (int w = 0; w < 640; w++){for (int c = 0; c < 3; c++){//tensor:chw排列,这里待转tensor为(1,3,640,640)int out_index = c * 640 * 640 + h * 640 + w;//mat是hwc排列,原始mat为(640,640,3)data1[out_index] = (float(img.at(h, w)[c])-127.5)/128.0;}}}infer_request.infer();

3.推理后结果提取:

 float* pdata_score = infer_request.get_output_tensor(n).data();  

使用opencv自带的dnn模块进行推理

0.模型构造

这里使用的是onnx

cv::dnn::Net net;this->net = cv::dnn::readNet(config.modelfile);

1.数据准备:原始mat变形为cv::dnn::Net的输入mat

cv::Mat img = this->resize_image(frame, &newh, &neww, &padh, &padw);cv::Mat blob;//先减127.5再调换RB通道,再除以128.0,得出的blob为(1,3,640,640)的形状cv::dnn::blobFromImage(img, blob, 1 / 128.0, cv::Size(this->inpWidth, this->inpHeight), cv::Scalar(127.5, 127.5, 127.5), true, false);this->net.setInput(blob);std::vector outs;this->net.forward(outs, this->net.getUnconnectedOutLayersNames());

2.推理后结果提取:

float* pdata_score = (float*)outs[n * 3].data;

相关内容

热门资讯

总算了解开挂!众乐乐开挂AI攻... 总算了解开挂!众乐乐开挂AI攻略,众乐乐多功能都是存在有开挂攻略1、上手简单,内置详细流程视频教学,...
程序员教你辅助!英豪娱乐城辅助... 程序员教你辅助!英豪娱乐城辅助AI工具,英豪娱乐城多功能原来真的是有辅助工具暗藏猫腻,小编详细说明英...
盘点几款辅助!冰球突破辅助AI... 盘点几款辅助!冰球突破辅助AI插件,冰球突破多功能其实真的有辅助插件1、冰球突破脚本辅助下载、冰球突...
分享一款辅助器!建湖好运麻将辅... 分享一款辅助器!建湖好运麻将辅助器AI教程,建湖好运麻将多功能竟然真的是有辅助器教程1、在建湖好运麻...
每日必备透视挂!新老夫子透视挂... 每日必备透视挂!新老夫子透视挂AI技巧,新老夫子多功能原来真的有透视挂技巧1、新老夫子脚本辅助下载、...
分享认知作弊器!星际互联作弊器... 分享认知作弊器!星际互联作弊器AI方法,星际互联多功能都是存在有作弊器方法1、每一步都需要思考,不同...
玩家必备攻略作弊器!闲逸斗地主... 玩家必备攻略作弊器!闲逸斗地主作弊器AI方法,闲逸斗地主多功能果然是有作弊器方法1、闲逸斗地主透视辅...
分辨真假透视挂!宁波游戏大厅透... 分辨真假透视挂!宁波游戏大厅透视挂AI软件,宁波游戏大厅多功能一直是有透视挂软件1、让任何用户在无需...
每日必看教程透视挂!君悦来透视... 每日必看教程透视挂!君悦来透视挂AI工具,君悦来多功能本来是有透视挂工具1、上手简单,内置详细流程视...
发现一款外挂!半岛外挂AI工具... 发现一款外挂!半岛外挂AI工具,半岛多功能一贯真的有外挂工具1、起透看视 半岛辅助软件价格2、随意选...