首页 > 编程知识 正文

ncnn使用,ncnn框架

时间:2023-05-05 11:05:53 阅读:232124 作者:3003

编写ncnn的C++步骤,基本分为五步:

1.定义模型:

ncnn:: Net  net;

2.load模型:

#由其他框架例如pytorch,tensorflow转化为ncnn的模型,包括两个文件.param和.bin

net.load_param("model.param");

net.load_model("model.bin");

3.定义输入数据:

cv::Mat bgr=cv::imread("path to image");

const int target_size = 300;//图像resize的大小

ncnn::Mat in=ncnn::Mat::from_pixels_resize(bgr.data,ncnn::Mat::PIXEL_BGR2RGB,bgr.cols,bgr.rows,target_size,target_size);

//归一化

const float mean_vals[3] = {123.675f, 116.28f, 103.53f};
const float norm_vals[3] = {1.0f, 1.0f, 1.0f};

in.substract_mean_normalize(mean_vals,norm_vals);

4.定义输出:

ncnn::Extractor ex=net.creator_extractor();

ex.set_num_threads(num_thread);//设置多线程

ex.input("input",in);

ncnn::Mat out;

ex.output("output",out);

5.后处理:

 std::vector<float>  cls_scores;

  // manually call softmax on the fc output
    // convert result into probability
    // skip if your model already has softmax operation
    {
        ncnn::Layer* softmax = ncnn::create_layer("Softmax");

        ncnn::ParamDict pd;
        softmax->load_param(pd);

        softmax->forward_inplace(out, shufflenetv2.opt);

        delete softmax;
    }

    out = out.reshape(out.w * out.h * out.c);

    cls_scores.resize(out.w);
    for (int j = 0; j < out.w; j++)
    {
        cls_scores[j] = out[j];
    }
 

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。