- 博客(4)
- 收藏
- 关注
RSS订阅原创 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
2840
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
2009
2
原创 C++调用pytorch模型（vs2015+libtorch+pytorch）
C++调用pytorch模型（vs2015+libtorch+pytorch） 参考网站 另一个网站：知乎平台 1.开发环境 vs2015+windows10 2.转换模型 pytorch的C++版本用的是Torch Script，官方给了两种将pytorch模型转成Torch Script的方法。这里主要介绍第一种Tracing。 3.准备工作 （1）在Pytorch官网下载所需要的Libtor...
2019-08-02 09:32:03
6965
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
1255
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人