天天看點

第一個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>
           

繼續閱讀