天天看点

第一个ncnn项目

之前一直想尝试ncnn项目,一致没有机会,最近找了一个项目练手。

是一个人像分割换背景的项目,先是pytorch训练pth模型,pth模型转onnx模型,onnx模型转ncnn模型。

然后是pth模型测试,onnx模型测试,ncnn模型测试。

pth模型和onnx模型测试结果一致,到ncnn时模型结果比较差。

pyotrch模型转pt模型使用libtorch预测效果跟pyotrch预测效果一致。ncnn预测结果有较大精度损失。

下面把ncnn预测代码列出:

#include "net.h"
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
           

继续阅读