pytorch
小小黑人
这个作者很懒,什么都没留下…
展开
-
C++调用libtorch新版本报错、出错
libtorch新的版本如何调用.pt文件一、在pytorch1.2版本中,我们使用Visual Studio时调用.pt文件时,会报错,如下1、不存在用户定义的从"torch::jit::script::Module"到"std::shared_ptr <>的适当转换。二、解决1、旧版本1.0调用2、Version1.13、Version 1.2三、详见github官网一、在pyt...原创 2020-02-09 13:37:21 · 2687 阅读 · 10 评论 -
pytorch+vs2015+win10
pytorch+vs2015+win10(一)安装GPU版本的pytorch一、前期准备二、基于Anaconda安装pytorch三、使用参考:[网页](https://zhuanlan.zhihu.com/p/54350088),非常感谢安装GPU版本的pytorch首先,查看自己电脑显卡的版本,右击“我的电脑”-“属性”-“设备管理器”-“显示适配器”,就可以看到自己的显卡。一些比较老的...转载 2019-07-31 22:45:09 · 1181 阅读 · 2 评论 -
C++调用pytorch模型(vs2015+libtorch+pytorch)
C++调用pytorch模型(vs2015+libtorch+pytorch)参考网站另一个网站:知乎平台1.开发环境vs2015+windows102.转换模型pytorch的C++版本用的是Torch Script,官方给了两种将pytorch模型转成Torch Script的方法。这里主要介绍第一种Tracing。3.准备工作(1)在Pytorch官网下载所需要的Libtor...原创 2019-08-02 09:32:03 · 6723 阅读 · 10 评论 -
制作自己的数据集
一、给自己的数据集打上标签,生成txt文件Eg:这里的图像样本已经分好类0-7共8类为文件夹内所有的图像打上标签源码参考import os''' 为数据集生成对应的txt文件'''train_txt_path = './train.txt'train_dir = './train/'#valid_txt_path = './valid.txt'#valid_d...原创 2019-08-31 10:27:28 · 1920 阅读 · 2 评论