pytorch
文章平均质量分 55
jdefla
欢迎留言讨论,博客私信回复不及时,sina weibo:@CD_555
展开
-
pytorch | 深度学习分割网络U-net的pytorch模型实现
这个是pytorch出来没多久的时候写的了,现在看是非常傻逼的方法,羞耻感十足。 推荐学习项目【pix2pix】的代码,优雅! –作者 2018.1.30U-Net 的实现现在github上非常多了吧!用dense-net大概也随随便便吊打了吧!不要用我这个啦~批判性参考一下pytorch咋用还差不多~! –作者 2018.4.09大概不支持pytorch 0.4以及以上版本...原创 2017-03-08 21:48:21 · 50033 阅读 · 26 评论 -
pytorch | 利用batch normalization对Variable进行normalize/instance normalize
天啦噜!!我发现更新的pytorch已经有instance normalization了!! 不用自己折腾了!! -2017.5.25利用 nn.Module 里的 子类 _BatchNorm (在torch.nn.modules.batchnorm中定义),可以实现各种需求的normalize。 在docs里,可以看到,有3种normalization layer,但其实他们都...原创 2017-04-01 18:14:59 · 35496 阅读 · 5 评论 -
pytorch | 计算图像的一阶导/梯度/gradient
[pytorch] 计算图像的一阶导 / 梯度 / gradient在图像转换任务中常见的total variation loss(tvloss,总变分,一般作为平滑的规则化项)需要对图像的梯度求平方和。 style-transfer系的github项目,tvloss求法如下:class TVLoss(torch.nn.Module): def __init__(self):...原创 2018-01-30 18:43:50 · 16619 阅读 · 8 评论