pytorch
TYUT_xiaoming
计算机视觉/深度学习(DL/CV)
展开
-
使用pytorch配合flask搭建简易web服务
视频作者:https://www.bilibili.com/video/BV1Qv41117SR?from=search&seid=15744399933120286124UP主:原创 2021-05-30 22:45:53 · 367 阅读 · 0 评论 -
Pytorch 使用不同版本的cuda
Nvidia cuda 与 cudatoolkit 的区别注意:确保编译CUDA版本和运行时CUDA版本匹配。你可以在PyTorch网站上查看支持的CUDA版本的预编译包。以下内容转载自:https://www.cnblogs.com/yhjoker/p/10972795.html在使用 Anaconda 安装 Pytorch 深度学习框架时,可以发现 Anaconda 会自动为我们安装 cudatoolkitconda install pytorch torchvision cud原创 2020-08-09 16:54:10 · 3162 阅读 · 1 评论 -
python os.system
os.system()用于运行其他程序或者脚本或者终端的命令示例:import oscmd_str = 'touch 1.txt'os.system(cmd_str)import oscmd_str = 'python phon_test.py'os.system(cmd_str)cmd_str = 'ffmpeg -f image2 -i {}/%05d.j...原创 2020-02-26 12:26:02 · 1095 阅读 · 0 评论 -
pytorch-图像预处理
参考链接:https://blog.csdn.net/qq_34199326/article/details/84072505img_ = img[:,:,::-1].transpose((2,0,1))img是【h,w,channel】,这里的img[:,:,::-1]是将第三个维度channel从opencv的BGR转化为pytorch的RGBtranspose((2,0,...原创 2019-11-26 17:37:06 · 741 阅读 · 0 评论 -
Pytorch view() permute() contiguous() transpose()
目录原始tensor1.view()2.permute()3.contiguous()4.transpose()5.squeeze() 和 unsqueeze()原始tensorimport torcha = torch.Tensor([[[1,2,3],[4,5,6],[7,8,9]]])print(a)print(a.size())输出:...原创 2019-11-27 17:34:26 · 1175 阅读 · 0 评论 -
ubuntu16.04 安装 Pytorch
pytorch网址:https://pytorch.org/torchvison网址:https://pypi.org/project/torchvision/0.2.0/Via pip从对应的网站下载whl文件# CPU-only buildhttps://download.pytorch.org/whl/cpu/torch_stable.html# CUDA 8....原创 2019-08-19 22:36:32 · 8185 阅读 · 2 评论 -
Pytorch学习笔记(一)-——《深度学习入门之pytorch》
pytorch链接http://pytorch123.com/https://pytorch-cn.readthedocs.io/zh/latest/目录1.tensor2.Variable (变量)3.nn.Module4.模型的保存和加载5.一个回归例子1.tensorTensor是一个多维矩阵 PyTorch的Tensor可以和numpy...原创 2019-08-29 23:33:21 · 319 阅读 · 0 评论 -
Pytorch学习笔记(二)-——《深度学习入门之pytorch》
简单的三层全连接神经网络改进网络:添加激活函数nn.Sequential() 函数将网络的层组合在一起改进网络:添加BN注意BN一般放在全连接层的后面、非线性层(激活函数)的前面训练一个网络导入一些相关的包数据预处理transforms.ToTensor():将图片转换成Tensor, 转换的过程中自动将图片标准化,Tensor的范围是(0,1)...原创 2019-09-02 22:08:38 · 215 阅读 · 0 评论 -
Pytorch学习笔记(三)-——《深度学习入门之pytorch》
目录卷积层提取层结构提取参数卷积层提取层结构提取网络中的某一层或者几层nn.Module的属性children() :返回下一级模块的迭代器(下图中只会返回到:layer1、layer2) named_children() :迭代器+网络层的名字modules() :返回模型中所有模块的迭代器,访问到最内层(可以访问到co...原创 2019-09-04 23:08:51 · 269 阅读 · 0 评论 -
Pytorch学习笔记(四)-——《深度学习入门之pytorch》
图像增强的方法原创 2019-09-12 23:51:06 · 183 阅读 · 0 评论 -
pytorch中x = x.view(x.size(0), -1) 的理解
在pytorch的CNN代码中经常会看到x.view(x.size(0), -1)首先,在pytorch中的view()函数就是用来改变tensor的形状的,例如将2行3列的tensor变为1行6列,其中-1表示会自适应的调整剩余的维度a = torch.Tensor(2,3)print(a)# tensor([[0.0000, 0.0000, 0.0000],# ...原创 2019-09-13 13:31:04 · 16214 阅读 · 5 评论