自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(6)
  • 收藏
  • 关注

原创 Pytorch神经网络结构可视化之tensorwatch

Tensorwatch查看网络结构图踩得坑踩坑日记出坑日记一、环境安装二、graphviz安装三、逐个解决错误如果你想使用tensorwatch绘制网络的结构图,然后绘制的网络图如下,那就是版本较高带来的问题!踩坑日记本来自己使用的pytorch版本是1.8,听说tensorwatch对版本有要求,特意装了一个低版本的pytorch,结果还是掉坑里了。pytorch=1.4torchvision=0.5.0pydot=1.4.2tensorwatch=0.9.1上述的版本可以画出图,但是画出

2021-07-03 15:19:41 2069 1

原创 pytorch中实现Balanced Cross-Entropy

当你明白了pytorch中F.cross_entropy以及F.binary_cross_entropy是如何实现的之后,你再基于它们做改进重新实现一个损失函数就很容易了。1、背景变化检测中,往往存在样本不均衡的情况,也就是changed pixel很少,而unchange pixel占据大多数,论文DASNet: Dual Attentive Fully Convolutional Siamese Networks for Change Detection in High-Resolution Sat

2021-04-16 21:00:37 4263 6

原创 一文搞懂F.binary_cross_entropy以及weight参数

相信有很多人在用pytorch做深度学习的时候,可能只是知道模型中用的是F.binary_cross_entropy或者F.cross_entropy,但是从来没有想过这两者的区别,即使知道这两者是分别在什么情况下使用的,也没有想过它们在pytorch中是如何具体实现的。在另一篇文章中介绍了F.cross_entropy()的具体实现,所以本文将介绍F.binary_cross_entropy的具体实现。当你分别了解了它们在pytorch中的具体实现,也就自然知道它们的区别以及应用场景了。1、pytor

2021-04-16 10:05:37 26563 16

原创 pytorch中F.cross_entropy和F.nll_loss的区别

搞懂函数的区别的最好方法就是产看函数的底层实现。首先可以在pytorch中分别点击进入函数内部,看他们的具体实现。1、F.cross_entropy的内部实现是:并且cross_entropy函数的返回值是:2、F.nll_loss的内部实现是:3、结论这里先上结论。这两种损失的区别在于input上的操作,对于cross_entropy来说,他首先会对input进行log_softmax操作,然后再将log_softmax(input)的结果送入nll_loss;而nll_loss的inpu

2021-04-07 11:31:57 13061 1

原创 一文搞懂F.cross_entropy中的weight参数

交叉熵是在分类任务中常用的损失函数,对于样本均衡的分类任务我们可以直接使用。但当我们面对样本类别失衡的情况时,导致训练过程中的损失被数据量最多的类别的主导,从而导致模型不能被有效的训练。我们需要通过为不同的样本损失赋予不同的权重以平衡不同类别间数据量的差异。这时了解一下F.cross_entrpy中的weight参数的底层是如何实现的,是非常有必要的!关于Pytorch中F.cross_entropy详细的实现过程请看添加链接描述这篇博客。下面分析weight是如何作用的。1、数据准备input =

2021-04-04 17:49:07 10480

原创 一文搞懂F.cross_entropy的具体实现

一文搞懂F.cross_entropy的计算过程交叉熵是在分类任务中常用的一种损失函数,本文详细介绍了pytorch中是如何实现交叉熵的细节!!!pytorch中的交叉熵函数为F.cross_entropy(input, target),本文以变化检测或语义分割中用到的数据模型为例:input的维度为[batchsize,classes,width,height],target的维度为[batchsize,width,height]。1. 随机生成模型数据input = torch.rand([1

2021-04-03 20:35:36 20888 17

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除