ncnn：读取二维数据

最新推荐文章于 2024-09-30 15:28:30 发布

baye_DOA

最新推荐文章于 2024-09-30 15:28:30 发布

阅读量572

点赞数 1

分类专栏： c++ 文章标签： c++ 算法开发语言

本文链接：https://blog.csdn.net/baye_DOA/article/details/128733103

版权

c++ 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

该代码示例展示了如何利用ncnn库进行二维数据的读取和处理，加载预训练模型进行年龄检测。它首先从文件中读取61x37的数据矩阵，然后将数据转换为ncnn::Mat对象输入模型进行预测。模型参数和模型权重分别从Simple_sim_opt.param和Simple_sim_opt.bin加载。

摘要由CSDN通过智能技术生成

Mat类型变量读取二维数据：

#include "net.h"
#include <fstream>
#include <iostream>
#include <chrono>
#include <stdio.h>
#include <vector>
#if defined(USE_NCNN_SIMPLEOCV)
#include "simpleocv.h"
#endif
using namespace std;

struct Object
{
    int label;
    float prob;
};

static int detect_age(std::vector<Object>& objects)
{

    ncnn::Net age;
//    age.opt.use_vulkan_compute = true;
    if (age.load_param("Simple_sim_opt.param"))
        exit(-1);
    if (age.load_model("Simple_sim_opt.bin"))
        exit(-1);

    float data[61][37] = { 0 };
    ifstream infile;
    infile.open("/home/1.txt");

    for (int i = 0; i < 61; i++){
        for (int j = 0; j < 37; j++){
            infile >> data[i][j];
        }
    }
    infile.close();

    // 存储数据：
    ncnn::Mat m(37,61);
    std::copy(data[0],data[61],(float *)m);

    ncnn::Extractor ex = age.create_extractor();
    ex.input("input", m);
    ncnn::Mat out;
    ex.extract("output", out);
    for (int i = 0; i < out.total(); ++i){
        std::cout<< out[i] << std::endl;
    }
    objects.clear();

    return 0;
}

int main()
{

//    test();
    std::vector<Object> objects;
    detect_age(objects);

    return 0;
}

直接用 ncnn::Mat m(37, 61, data[0]) 就可以了，不需要 std::copy