| |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
-> 人工智能 -> 利用LIbtorch1.9实现在VS2019中调用孪生网络模型 -> 正文阅读 |
|
[人工智能]利用LIbtorch1.9实现在VS2019中调用孪生网络模型 |
前言此前,已经分享了
VS2019使用C++调用并部署pytorch VGG模型全过程(libtorch+opencv)(cpu+gpu). VS2019使用Darknet调用YOLOV3模型并测试(CPU+GPU). 在实际项目过程中,又需要在VS中调用孪生网络模型,在网上查阅一番,发现并没有相关资料,所以只有自食其力/(ㄒoㄒ)/~~。在了解模型输入数据的底层相关处理代码后,解决了这一问题,故简单做个小记录,大家可以参考一下,😀 说明本博客主要讨论如何针对孪生网络这样需要输入两张图片进行预测的网络模型,进行模型序列化(.pth->.pt)以及在vs项目中实现基本运算。 方法序列化模型:1.设置网络模型
2.生成一个随机输入维度的输入 输入shape应满足网络要求
注意:创建好张量后需要为其在最外层增加一个维度 3.将两张量在新增维度上进行拼接
4.利用torch.jit.trace序列化模型
WHY? 跳转至网络内部代码,并调试可以发现真正进入网络的数据实际有5个维度。第一个维度的值为2,说明待比较的图片张量在此维度进行了拼接操作。故正确的输入数据应作如上处理。😀 VS调用模型预测:主要介绍下输入数据在vs端的处理操作,模型调用等步骤与VGG模型一致,详见VS2019使用C++调用并部署pytorch VGG模型全过程(libtorch+opencv)(cpu+gpu).
2.设置两个输入张量并扩充维度
3.拼接张量
4.创建输入变量,并将张量input置入
5.模型预测
-----------tbc---------- 【作者有话说】 |
|
|
上一篇文章 下一篇文章 查看所有文章 |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
360图书馆 购物 三丰科技 阅读网 日历 万年历 2025年1日历 | -2025/1/9 1:23:13- |
|
网站联系: qq:121756557 email:121756557@qq.com IT数码 |