pytorch
Geek_of_csdn
是菜就要认,挨打要立正。一个奋斗的菜鸟,目前在深圳某公司工作。本账号下所有上传的资料都可以直接通过邮箱免费索要,邮箱地址:dianxin2wu@foxmail.com
展开
-
Pytorch:利用预训练好的VGG16网络提取图片特征
前言这里的提取图片特征特指从VGG网络的最后一个conv层进行提取。虽然下面代码里面给出的是VGG16作为例子,其实也可以用其他的已经经过训练了的神经网络,包括自己训练的。代码模型结构相关基本知识首先说下加载模型,这里用的是torch官方提供的已经训练好的模型,只需要从torchvision模块导入:import torchvision.models as modelsmodel =...原创 2019-02-24 09:43:15 · 62178 阅读 · 104 评论 -
【小技巧】生成除指定数字外的随机整数
from random import choiceprint(choice([i for i in range(0,9) if i not in [2,5,7]]))如果是生成一个list:from random import choicetmp = []for j in range(10): tmp.append(choice([i for i in range(0,9) if...原创 2018-12-21 00:32:44 · 2889 阅读 · 0 评论 -
关于贫僧在使用PyTorch时遇到了runtime error(59):device-side assert triggered at XXX这样神奇的BUG的事
标题里的XXX是省略之后的文字的意思,不是真的“XXX”。某天贫僧在训♂练模型的时候遇到了这个错误提示。其实这个错误提示很模糊(非常模糊),基本上是没有办法定位到出错的地方的,在查找资料之后发现了一下解决方法(在这里总结下):1. 改为在CPU上运行这是因为当模型在GPU上运行的时候其实是没办法显示出真正导致错误的地方的(按照PyTorch Dev的说法:“Because of the as...原创 2019-01-17 16:39:15 · 31807 阅读 · 9 评论 -
【小技巧】关于pytorch中拥有batch的矩阵的相加
应用场景将两个batch_size = 100的矩阵相加时,如果维度不完全相同,例如100×512100 \times 512100×512与100×196×512100 \times 196 \times 512100×196×512大小的矩阵相加时,因为pytorch不能够直接进行broadcasting,所以如果直接a + b相加的话会直接报错(如果没有batch_size的话则可以正常进...原创 2019-01-18 20:01:06 · 8508 阅读 · 0 评论 -
PyTorch:利用预训练好的ResNet-152网络提取图片特征
这篇博文相当于是对上一篇博文Pytorch:利用预训练好的VGG16网络提取图片特征 的补充,本文中提到的提取方式与前文中的不同。另外,因为TorchVision提供的训练好了的ResNet效果不好,所以本文中将会使用由ruotianluo提供的从Caffe转换过来的ResNet模型(具体可以看这个repo,如果好奇怎么转换的话)。代码以下代码节选自pytorch-vqa的preproces...原创 2019-05-13 19:16:59 · 18796 阅读 · 11 评论