C++推理ncnn模型报错Segmentation fault (core dumped)

原创

已于 2022-07-02 21:23:56 修改 · 1.5k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #开发语言

于 2022-06-29 16:28:05 首次发布

在执行C++代码进行ncnn模型推理时遇到Segmentation fault (core dumped)的问题。尽管已确认core文件大小无限制，但起初并未生成core文件。通过安装gdb并定位core文件，发现在将ncnn::mat转换为cv::mat的过程中出现错误，原因是ncnn推理结果包含nan值，导致转换失败。解决办法是检查和修正输入数据的归一化步骤。

推理代码如下：

#include "net.h"
#include <iostream>
#include <fstream>
#include <algorithm>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <stdio.h>
#include <vector>
#include <opencv2/opencv.hpp>

using namespace std;

void pretty_print(const ncnn::Mat& m)
{
   
   
    for (int q=0; q<m.c; q++)