赞
踩
之前一直想尝试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>
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。