mujiangyao的博客

继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

由于毕业设计的设计，学习使用了一下libtorch的使用。 主要参考链接https://zhuanlan.zhihu.com/p/92374964。 在写好cmake文件以及基本的加载代码之后，调用模型对单张图片进行深度预测，并将结果保存下来。代码如下： #include <torch/script.h> #include <iostream> #include <string> #include <memory> #include <ope..

在使用tensorboard的时候遇到了很多问题，特此记录一下。 tensorboard运行时提醒找不到库 如果使用的是anaconda，则要查看一下anaconda的bin是否添加到路径中，否则怎么安装都会报错。我反复使用conda安装了tensorflow以及tensoroardx，都不起作用。最后使用pip安装以及添加了bin的路径解决了找不到模块的错误。 tensorboard打开为白板 参考了此篇博文解决了问题：tensorboard命令打开空白页面的解决办法 ...