深度学习
文章平均质量分 71
ncc1995
这个作者很懒,什么都没留下…
展开
-
【深度学习】Python数据集整理——基础知识(dict.fromkeys()、random的使用)
Python字典 (dictionary) fromkeys() 方法描述:Python 字典 fromkeys() 函数用于创建一个新字典,以序列seq中元素做字典的键,value为字典所有键对应的初始值。语法:dict.fromkeys(seq[], value)参数:seq:字典键值列表。value:可选参数,键值对应的值。实例:list = [...原创 2019-03-15 22:18:01 · 1965 阅读 · 0 评论 -
【PyTorch】RuntimeError: one of the variables needed for gradient computation has been modified by an
错误:RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation报错原因:import torchx = torch.FloatTensor([[1., 2.]])w1 = torch.FloatTensor([[2.], ...原创 2019-08-14 09:57:11 · 76981 阅读 · 16 评论 -
【PyTorch】cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generated/.
记得上次遇到过一次该错误,当时没有总结,今天又遇到了,总结一下!问题:cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generated/../generic/T HCTensorMathPointwise.cu:265解决方案:...原创 2019-08-13 18:07:11 · 1019 阅读 · 0 评论 -
【PyTorch】AttributeError: Can't get attribute 'BCNN'
错误:AttributeError: Can't get attribute 'BCNN' on <module '__main__' from '今天在服务器上训练好一个网络,用torch.save(model, 'net.pkl')将模型和参数一并保存了起来,方便在自己电脑上测试。把保存的模型下载下来进行测试的时候发现报了上述错误。原来保存下来的模型和参数不能在没有类定义时直...原创 2019-08-13 17:55:33 · 8537 阅读 · 3 评论 -
【TensorFlow】读取ckpt文件中的tensor,将ckpt模型转为pytorch模型
今天想用MobileNet V1训练自己的数据,发现pytorch没有MobileNet V1的预训练权重,只好先下载TensorFlow的预训练权重,再转成pytorch模型。1、读取ckpt中的Tensor名称以及Tensor值TensorFlow的MobileNet V1预训练权重文件如下:解压完文件后,发现没有.ckpt文件,文件名只需'./my_model/mobile...原创 2019-06-29 16:18:31 · 6813 阅读 · 2 评论 -
【深度学习】1x1卷积核的作用
1×1卷积的作用以下内容摘抄自:http://www.caffecn.cn/?/question/136问:发现很多网络使用了1X1卷积核,这能起到什么作用呢?另外我一直觉得,1X1卷积核就是对输入的一个比例缩放,因为1X1卷积核只有一个参数,这个核在输入上滑动,就相当于给输入数据乘以一个系数。不知道我理解的是否正确答1:对于单通道的feature map和单个卷积核之间的卷积来说,...转载 2019-06-28 21:46:19 · 304 阅读 · 0 评论 -
【深度学习】MobileNet参数数量计算,参数量减少
正常参数量与mobileNet参数量的对比:MobileNet的网络结构:原创 2019-06-28 20:53:34 · 5619 阅读 · 1 评论 -
【PyTorch】pytorch加载自己的数据集,重写Dataset类
len方法,提供了dataset的大小;getitem方法, 该方法支持从 0 到 len(self)的索引。#filenames是训练数据文件名称列表,labels是标签列表class MyDataset(Dataset): def __init__(self, filenames, labels, transform): self.filenames ...原创 2019-06-28 11:15:42 · 4626 阅读 · 1 评论 -
【pyTorch】nn.Conv2d中的padding计算
今天写GoogLeNet网络的时候,计算网络参数,查了一下卷积中padding的计算方式: H_{out} = \left\lfloor\frac{H_{in} + 2 \times \text{padding}[0] - \text{dilation}[0] \times (\text{kernel_size}[0...原创 2019-06-26 21:24:46 · 3267 阅读 · 1 评论 -
【PyTorch】数据增强之Label shuffling实现(神经网络训练时样本不平衡解决方法)
前面的数据是采用PyTorch的ImageFolder读取的,读取后的数据进行了训练集数据集的划分,分别得到了image路径列表和label标签列表,再传入自己写的label_shuffling进行数据增强,返回后通过PyTorch的DataLoader加载到网络里。def label_shuffling(data, labels): #得到样本数最多的类别的样本数 maxN...原创 2019-06-28 11:16:16 · 899 阅读 · 1 评论 -
【TensorFlow】batch normalization的TensorFlow实现
TensorFlow中batch normalization的实现有两种,一种是tf.layers.batch_normalization,一种是tf.nn.batch_normalization,在此处,我使用的是第二种。原理就不贴了。代码如下:def BN(input, isTraining=False, name='BatchNorm', moving_decay=0.9, e...原创 2019-05-16 20:43:39 · 303 阅读 · 0 评论 -
【TensorFlow】maxout实现以及TensorFlow各种基础知识
一、TensorFlow权重初始化截断高斯分布下面的初始化改进都是基于保证输出与输入的分布相差不大的思想。xavier初始化(适用于激活函数是sigmoid和tanh)MSRA初始化(适用于激活函数relu)二、TensorFlow参数初始化tf.global_variables_initializer三、tf.summaryscalar(name, ...原创 2019-03-22 17:20:01 · 1184 阅读 · 0 评论 -
【深度学习】运行grad-cam(pytorch代码)遇到的一些问题
用原代码跑就没有问题,因为要提取自己数据集的判别性特征,而数据集加载必须使用pytorch自带的DataLoader。这时候最后可视化热力图的时候图片明显不对,下面就是一点一点排除错误的过程。最后是直接将二值化后的mask找出坐标,裁剪出具有判别性的区域了!...原创 2019-08-14 10:23:47 · 2958 阅读 · 7 评论