- 博客(6)
- 资源 (6)
- 收藏
- 关注
RSS订阅原创 c++判断string字符串是否含有某个子串
1、string类函数findC++的string类提供了字符串中查找另一个字符串的函数find。其重载形式为：string::size_typestring::find(string &);功能为在string对象中，查找参数string类型的字符串是否存在，如果存在，返回起始位置。不存在则返回string::npos。#include <iostream&g...
2019-06-14 14:45:59
31998
原创 pytorch模型格式转换
将生成的ckpt_e_50.pth文件转为适合在pc端做推断的.pt文件：model = UNet(3, 1)modelname = 'ckpt_e_50.pth'ckpt = torch.load(opt.pretrain + modelname)model.load_state_dict(ckpt['state_dict'], strict=False)model.eval()...
2019-06-11 18:27:23
13754
12
原创 c++调用pytorch模型进行推断
当训练好一个CNN模型之后，可能要集成到项目工程中，或者移植到到不同的开发平台(比如Android, IOS)， 一般项目工程或者App大多数采用C/C++, Java等语言，但是采用pytroch训练的模型用的是python语言，这样就存在一个问题，如何使用C/C++调用预训练好的模型, 如果解决了这个问题，那么训练好的模型才可以在App中得到广泛应用。PyTorch模型从Python到C+...
2019-06-10 16:23:32
9564
9
原创 vs2015深度学习推断报错
问题：不涉及深度学习框架的c++代码，平时编译运行都没有问题；但是涉及深度学习的c++代码，如tensorflow、libtorch，同样的代码，在其他电脑上可正确编译、运行，但在本机编译失败。。。心路历程。。。刚开始以为是libtorch库的问题，分别下载cuda版本和非cuda版本，release版和debug版进行配置测试，采用直接配置和cmake的方式均出现相同的报错。。...
2019-06-10 11:07:05
271
原创 win10安装cuda9.0，cudann
CUDA是NVIDIA推出的用于自家GPU的并行计算框架，也就是说CUDA只能在NVIDIA的GPU上运行，而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。cuDNN（CUDA Deep Neural Network library），是NVIDIA打造的针对深度神经网络的加速库，是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型，cuDNN不是必须的，但...
2019-06-06 18:33:26
24086
1
原创 c++对vector中的元素进行随机排序
头文件：#include <iostream>std::vector<std::string> filenames;进行赋值操作之后，打乱向量中的顺序：std::random_shuffle(filenames.begin(), filenames.end());
2019-06-06 16:12:47
7209
2
mars-small128.zip
2019-11-25
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人