![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
人工智能
mujiangyao
这个作者很懒,什么都没留下…
展开
-
libtorch 的使用 - CUDA版本
继libtorch的初步使用后,更新一版CUDA版本的代码。main函数#include <torch/script.h>#include <iostream>#include <string>#include <memory>#include <opencv2/opencv.hpp>using namespace std;int main(int argc, const char* argv[]) { if(ar原创 2021-01-13 11:30:58 · 2873 阅读 · 0 评论 -
libtorch的初步使用
由于毕业设计的设计,学习使用了一下libtorch的使用。主要参考链接https://zhuanlan.zhihu.com/p/92374964。在写好cmake文件以及基本的加载代码之后,调用模型对单张图片进行深度预测,并将结果保存下来。代码如下:#include <torch/script.h>#include <iostream>#include <string>#include <memory>#include <ope..原创 2021-01-07 14:37:47 · 864 阅读 · 0 评论 -
tensorboard使用中的一些问题记录
在使用tensorboard的时候遇到了很多问题,特此记录一下。tensorboard运行时提醒找不到库如果使用的是anaconda,则要查看一下anaconda的bin是否添加到路径中,否则怎么安装都会报错。我反复使用conda安装了tensorflow以及tensoroardx,都不起作用。最后使用pip安装以及添加了bin的路径解决了找不到模块的错误。tensorboard打开为白板参考了此篇博文解决了问题:tensorboard命令打开空白页面的解决办法...原创 2020-12-25 14:30:36 · 135 阅读 · 0 评论