Zookeeper 国外镜像 java设计模式 vue组件 数据算法 5G arrays animation uicollectionview vue中文网 vue教程入门 网页后台模板 jquery解析json数据 mysql在线测试 vim跳到文件末尾 javaweb是前端还是后端 python界面 javafile java斐波那契数列 java正则表达式匹配 java如何连接mysql java八大基本数据类型 java多线程编程 java的date linux密码 战地2单机地图 高等数学同济第七版 日历制作模板 ipad锁屏 unity3d下载 视频加字幕软件 0x000008e 分屏软件 Linux进程查看 祸星龙 强制换行快捷键 vue数组更新 mdb是什么文件 win7显卡驱动下载 工行天天益
当前位置: 首页 > 学习教程  > 编程语言

c++调用pytorch训练出来的模型

2020/7/24 10:19:42 文章标签:

方法一:利用pytorch的c++版本的API:libtorch

方法二:利用opencv的readNetFromTorch(),导入的模型必须是 torch.save() 方法保存的文件。且opencv此函数只能读取序列化 nn.Module 对象,例如:
nn.Sequential
nn.Parallel
nn.Concat
nn.Linear
nn.SpatialConvolution
nn.spatialMaxPooling, nn.SpatialAveragePooling
nn.ReLU, nn.TanH, nn.Sigmoid
nn.Reshape
nn.SoftMax, nn.LogSoftMax
如果pytorch网络有基于nn.Module的自定义对象,则无法使用,opencv不能读取

除此之外,opencv还有readNetFromTensorflow(),readNetFromCaffe(),readNetFromONNX()等

ONNX是一个各种框架的的一个转换格式,相当于一个桥梁,使得模型可以在tensorflow、pytorch、caffe、MXNet之间互相转换。


本文链接: http://www.dtmao.cc/news_show_50239.shtml

附件下载

相关教程

    暂无相关的数据...

共有条评论 网友评论

验证码: 看不清楚?