c++ 调用mnn

谁践踏了优雅 2023-01-21 02:12 278阅读 0赞

之前写了一篇文章训练了一个模型，并转换成mnn模型 ，并且使用了python进行调用

不过并没有使用c++进行调用

模型还是使用那篇文章的模型，调用代码如下:

#include <iostream>
    #include<opencv2/core.hpp>
    #include<opencv2/imgproc.hpp>
    #include<opencv2/highgui.hpp>
    #include<MNN/Interpreter.hpp>
    #include<MNN/ImageProcess.hpp>
     
    using namespace std;
    using namespace cv;
    using namespace MNN;
    int main()
    {   auto net = std::shared_ptr<MNN::Interpreter>(MNN::Interpreter::createFromFile("E:\\vs2019\\first\\first\\model-flow.mnn"));//创建解释器
        cout << "Interpreter created" << endl;
        ScheduleConfig config;
        config.numThread = 8;
        config.type = MNN_FORWARD_CPU;
        auto session = net->createSession(config);//创建session
        cout << "session created" << endl;
        //读取图片
        cv::Mat image = cv::imread("C:\\Users\\Administrator\\Pictures\\397.jpg");
        cv::Mat img_resized;
        cv::resize(image, img_resized, cv::Size(180,180));
     
        auto inTensor = net->getSessionInput(session, NULL);
        auto outTensor = net->getSessionInput(session, NULL);
        auto _Tensor = MNN::Tensor::create<float>(inTensor->shape(), NULL, MNN::Tensor::TENSORFLOW);
        //cout << _Tensor->elementSize() << endl;
        for (int i = 0; i < _Tensor->elementSize(); i++) {
            _Tensor->host<float>()[i] = image.data[i];
        }
        inTensor->copyFromHostTensor(_Tensor);
        inTensor->printShape();
        //cout << *(inTensor->host<float>()+1) << endl;
        //_Tensor->print();
        _Tensor->printShape();
        //推理
        net->runSession(session);
        auto output= net->getSessionOutput(session, NULL);
        MNN::Tensor feat_tensor(output, output->getDimensionType());
        output->copyToHostTensor(&feat_tensor);
        feat_tensor.print();
        waitKey(0);
        return 0;
    }

————————————————  
版权声明：本文为CSDN博主「喝粥也会胖的唐僧」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。  
原文链接：https://blog.csdn.net/zhou\_438/article/details/111869497