我也要用- 博客(10)
- 收藏
- 关注
RSS订阅原创 [pytorch]可视化feature map
[pytorch]可视化feature map可视化代码：transform函数：numpy转为PIL:tensor转为PIL：训练过程中调用可视化函数直接load预训练好的model并输出feature map在计算机视觉的项目中，尤其是物体分类，关键点检测等的实验里，我们常常需要可视化中间的feature map来帮助判断我们的模型是否可以很好地提取到我们想要的特征，进而帮助我们调整模型或者...
2020-02-24 18:48:54
2481
1
原创 [pytorch] torch.optimizer.lr_scheduler调整学习率
[pytorch] torch.optimizer.lr_scheduler调整学习率torch.optim.lr_scheduler.LambdaLRtorch.optim.lr_scheduler.StepLRtorch.optim.lr_scheduler.MultiStepLRtorch.optim.lr_scheduler.ExponentialLRtorch.optim.lr_sch...
2020-02-14 21:17:05
515
原创 [pytorch]训练的时候固定部分层的参数
[pytorch]训练的时候固定部分层的参数有些时候我们在写自己的网络的时候需要用到其他人或者pytorch的torchvision.models里预训练好的模型，但是我们可能希望固定一部分层的参数，在训练的时候不更新这些层的参数，这意味着我们希望反向传播计算梯度时，只计算剩余层的参数的梯度。我们知道，网络中的所有操作对象都是Variable对象，在这篇文章中笔者将介绍如何利用Variable...
2020-02-12 19:01:04
1507
8
原创 [pytorch] 使用tensorboardX可视化训练过程
[pytorch] 使用tensorboardX可视化训练过程安装tensorboardX使用tensorboardX创建writer实例使用add方法添加记录添加数字添加图片添加运行图添加高维嵌入向量其他在训练神经网络时，我们希望能更直观地了解训练情况，包括损失曲线、输入图片、输出图片、卷积核的参数分布等信息。这些信息能帮助我们更好地监督网络的训练过程，并为参数优化提供方向和依据。最简单的办法...
2020-02-06 09:43:50
692
原创 [pytorch]实现一个自己个Loss函数
[pytorch]实现一个自己个Loss函数pytorch本身已经为我们提供了丰富而强大的Loss function接口，详情可见Pytorch的十八个损失函数，这些函数已经可以帮我们解决绝大部分的问题，然而，在具体的实践过程中，我们可能发现还是存在需要自己设计Loss函数的情况，下面笔者就介绍一下如何使用pytorch设计自己的损失函数。Loss类的实现具体实践过程中，往往需要把loss ...
2020-02-03 11:42:16
1294
原创 [pytorch] Tensor, numpy与PIL格式的相互转换
[pytorch] Tensor, numpy与PIL格式的相互转换PIL与TensorTensor与Numpy图片展示多张图片的转换我们一般在pytorch或者python中处理的图像无非这几种格式：PIL：使用python自带图像处理库读取出来的图片格式Numpy：使用python-opencv库读取出来的图片格式Tensor：pytorch中训练时所采取的向量格式（注意，之后的讲...
2020-02-02 23:32:24
1463
原创 [pytorch]简单CNN网络在数据集MNIST上的实现
[pytorch]简单CNN网络在数据集MNIST上的实现数据集的下载与使用简单CNN网络的设计超参设置网络训练与测试模型加载训练过程最后学习机器学习的小伙伴接触到第一个神经网络模型往往就是最简单的CNN网络，这个网络可能只有一些卷积层、池化层、激活函数和全连接层组成，但就是这样一个简单的网络其实其实在一些分类问题上已经能取得不错的效果。同样，接触到的第一个数据集也可能就是著名的手写数字数据集M...
2020-02-02 22:38:21
178
原创 [pytorch]构建并加载自己的数据集
[pytorch]构建并加载自己的数据集)pytorch为我们封装好了很多经典的数据集在torchvision.datasets包里, torchvision.datasets这个包中包含MNIST、FakeData、COCO、LSUN、ImageFolder、DatasetFolder、ImageNet、CIFAR等一些常用的数据集，并且提供了数据集设置的一些重要参数设置，可以通过简单数据集设...
2020-02-02 17:18:11
863
1
原创 [python] 用pickle保存数据
【python】用pickle保存数据pickle数据的保存pickle数据的提取在训练模型的时候，如果我们想要保留中间层的结果，比较常用的是以字典的形式保存，python为我们提供了一个很强大的库pickle来帮助我们实现这一功能。pickle数据的保存pickle可以实现保存字典和列表等数据结构，使用的方法非常简单。import pickle a_dict = {'da': 111...
2020-02-01 23:08:04
180
原创 [pytorch] torch.nn.Conv3D 的使用介绍
[pytorch] torch.nn.Conv3D 的使用介绍torch.nn.Conv3D 参数输入参数输出参数网络参数使用示例torch.nn.Conv3D 参数3D卷积, 一般是在处理的视频的时候才会使用，目的是为了提取时序信息(temporal feature)，输入的size是(N,Cin,D,H,W)，输出size是(N,Cout,Dout,Hout,Wout)输入参数N: ...
2020-02-01 16:00:02
2597
5
空空如也
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人 TA的粉丝