mnn在c++中的推理demo(mnn不支持5维输入,即暂时不支持3dcnn的推理)

准备工作
1.使用cmake-gui在windows下编译mnn得到MNN.dll和MNN.lib(Release x64)
2.opencv的dll,lib,头文件可以在https://github.com/opencv/opencv/releases/tag/4.3.0的opencv-4.3.0-vc14_vc15.exe直接安装后得到,不用编译(opencv_world430.dll,opencv_world430.lib)
3.在vs2017中include mnn和opencv的所需要头文件(mnn缺什么直接在mnn的文件夹里搜完了把目录加到c++的include目录即可)以及lib,并将dll文件放入c++项目目录下面,设置环境为Release x64
4.将pytorch模型在linux下转onnx转mnn(linux下convertor比较好编译)

cpp代码

#include "Interpreter.hpp"
#include <opencv2/opencv.hpp>
#include <iostream>
#include "ImageProcess.hpp"

//这里执行的为一个分类模型
int main()
{
   
	const std::string mnn_path = "...\\model.mnn";
	std::shared_ptr<MNN::Interpreter> my_interpreter = std::shared_ptr<MNN::Interpreter>(MNN::Interpreter::createFromFile(mnn_path.c_str()));
	
	//config 
	MNN::ScheduleConfig config;
	int num_thread = 4;
	config.numThread = num_thread;
	MNN::BackendConfig backendConfig
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值