
pytorch
Jie Ou
学习
展开
-
pytorch loss 出现nan,原因之一,计算图中存在torch.sqrt或者 **0.5,以及如何解决
今天写一个loss函数dist=torch.sqrt(x*x+y*y)loss=soomthL1loss(dist,gt_dist)我随便写的几句示意代码,这样会导致在第一个iteration之后出现nan,第一次iteration之内,还是可以看到loss不为nan的。解决办法: 1、不开方,因为开方的求导会出现在分母上,因此需要避免分母为0! 2、tor...原创 2019-11-11 15:08:14 · 8555 阅读 · 3 评论 -
【简单的Pytorch回归模型案例】CNN去除随机噪声--修复2d高斯分布【pytorch demo】
一、这是个Pytorch学习案例,可以根据这个案例写自己的模型二、代码1、导入相关模块import torch from torch import nnimport torchvisionimport numpy as npimport cv2%matplotlib inlineimport matplotlib.pyplot as plt from torch....原创 2019-10-30 15:52:09 · 2225 阅读 · 0 评论 -
【目标检测】【数据增广】Data Augmentation For Object Detection
这是一个大佬在github 开源的项目,专门针对“目标检测”的“数据增广”!代码地址:https://github.com/Paperspace/DataAugmentationForObjectDetection博客地址:https://blog.paperspace.com/data-augmentation-for-bounding-boxes/如果我这样侵权了,请联系我删除!!...翻译 2019-01-23 16:17:44 · 7240 阅读 · 10 评论 -
【CSV to VOC 格式】智能盘点—钢筋数量AI识别 数钢筋,数据集转换 成我们常用格式“xml” code 代码
一、首先建立文件夹VOC_GJ在这个下面建立三个JPEGImages、Annotations、ImageSetsImageSets下面建立Main二、生成train.txt test.txtimport ostrain_file=open('path/ImageSets/Main/train.txt','w')test_file=open('path/ImageSets/...原创 2019-01-11 15:48:53 · 4110 阅读 · 28 评论 -
修改中。。。。【AIChallenger 2018 】【眼底水肿病变区域分割】冠军解决方案
内容完善中。。。赛道:眼底水肿病变区域分割队名:965728310比赛地址:https://challenger.ai/competition/fl2018,答辩视频地址可关注官网最新动态特别说明:未经作者同意以下内容请勿转载,或挪作他用!如果一下图片内容存在侵权,请及时联系作者删除!代码:后期更新题目内容:对OCT图像中,水肿类型进行多分类,且对病变区域进行分割 ...原创 2018-12-25 20:57:32 · 2560 阅读 · 20 评论 -
【Focal Loss】简单理解 及 Pytorch 代码 Focal Loss for Dense Object Detection
一、首先回顾下“交叉熵loss Cross Entropy Loss” CE(Pi)=-log(Pi)二、一般地说,我们数据集会存在类别不平衡问题,很多人会在loss上对应不同类别设置不同系数 loss就变成了上面的样子三、Focal loss其实就是通过数学公式上的改变,扩大了不平衡因素在loss上的影响...原创 2019-01-10 17:19:25 · 7075 阅读 · 6 评论 -
【SSD修改输入尺寸】how to change SSD input size?SSD640/BRFSSD_640
再也不局限于SSD300 和SSD512这两个版本了!!!BRFSSD_vgg实现修改input size!!!1、首先准备好一份SSD代码推荐使用:https://github.com/lzx1413/PytorchSSD框架:pytorch0.4.0运行遇到报错:关于Sample.py 下面什么说东西不能放在gpu之类的。进行如下修改anaconda3/envs/pyt...原创 2018-12-23 21:35:07 · 5704 阅读 · 21 评论 -
【single stage 多人姿态估计】Associative Embedding: End-to-End Learning for Joint Detection and Grouping
出处:NIPS2017核心内容:Single Stage 多人姿态估计文章摘要:这篇文章,提出了一个与之前所有方法都不一样的新的姿态估计方法,也可以用于instance 分割,目标检测等。这个方法是自下而上的,在此之前比较有名的自下而上的姿态估计领域的文章,是“OpenPose”-CMU。核心思想:先检测全图所有的关键点,如果是COCO数据集那么Heatmaps就有17+1个,如果...原创 2018-12-20 15:18:12 · 1773 阅读 · 0 评论 -
【深度学习总结】【深度学习-2018-年度总结】【计算机视觉】【预训练模型库】【神经网络技巧】【高性能计算】【人工智能】
【2018-新概念-深度学习-年度总结】:一、RFB模块文章地址:https://arxiv.org/abs/1711.07767 CVPR2017,虽然是2017年的,但是我还是要说代码地址:https://github.com/ruinmessi/RFBNet推荐代码地址:https://github.com/lzx1413/PytorchSSD核心思想:1、模仿人...原创 2018-12-21 11:17:29 · 721 阅读 · 0 评论 -
【深度学习总结】【深度学习-2018-年度总结】【计算机视觉】【预训练模型库】【神经网络技巧】【高性能计算】【人工智能】
时间:2018/12/20深度学习热潮:4年深度学习框架: 图片来源:http://m.elecfans.com/article/798648.html Pytorch可以说是用户数量提升最快的一个框架,上图来自于某位大佬的博客,时间节点2018年9月。现在pytorc...原创 2018-12-21 11:18:09 · 379 阅读 · 0 评论 -
python matplot plt画折现/曲线图,并在上面标记
##x,y 可以是list也可以是numpy格式,如x=[1,2,3,4,5,6]fig=plt.figure(figsize=(15,10))ax = fig.gca()for i in range(len(ans)): circle2 = plt.Circle((ans_x[i],ans_y[i] ), 3, color='blue') ax.add_artist(c...原创 2018-12-03 20:16:06 · 14567 阅读 · 0 评论 -
pytorch 如何设置 可学习参数
如何根据自己需求设定,可学习参数,并进行初始化。#比如cnn输出4个东西,你又不想concate到到一起,你想用权重加法,权重又不想自己设定,想让网络自己学#requires_grad=True这个很重要#设置前置网络及 可学习参数self.cnn=cnn_output4()self.fuse_weight_1 = torch.nn.Parameter(torch.FloatTe...原创 2018-11-27 16:14:53 · 14459 阅读 · 29 评论 -
pytorch 如何在预训练模型的 input上增减通道
如何把imagenet预训练的模型,输入层的通道数随心所欲的修改,从而来适应自己的任务#增加一个通道w = layers[0].weightlayers[0] = nn.Conv2d(4, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)layers[0].weight = torch.nn.Param...原创 2018-11-27 15:47:59 · 5841 阅读 · 17 评论 -
pytorch 如何以最方便的方式固定前置网络(特征提取网络)的参数
已经训练好了cnn1,现在想要用cnn1输出的结果来训练cnn2,但是要完全 固定cnn1的参数:self.step1=cnn1()for p in cnn1.parameters(): p.requires_grad = Falseself.init_=torch.load('cnn1.weight.path.pth.tar')self.step1.load_state...原创 2018-11-27 15:34:57 · 3287 阅读 · 0 评论 -
如何用基于resnet的Unet进行图像分割 基于Pytorch-0.5版本
1、关于Unet Unet主要用于医学图像的很多论文中,以及Kaggle竞赛和一些其他竞赛中“少类别”的图像分割。从我做实验的经验来说,像VOC这种类别比较多的分割任务,不容易收敛,效果较为差。2、Resnet34 我们的encode部分选择resnet34,decode部分为每一个block制作三层卷积,其中每个的第二层为upsample(bilinear/de...原创 2018-10-22 16:06:34 · 23070 阅读 · 24 评论 -
pytorch 小例子 demo 使用demo
我使用anaconda3,建议大家也用这个,好用!!!假设你安装完了anaconda3.一、创建一个测试Demo的实验环境conda create -n PyTorch-demo python=3.6The following NEW packages will be INSTALLED: ca-certificates: 2018.03.07-0 certi...原创 2018-08-03 21:40:31 · 9727 阅读 · 0 评论 -
pytorch master + anacoda3 + cudnn7+cudnn5+ 虚拟环境
首先说一下cudnn 不同版本是可以并存的, cudnn5系列要求在/usr/local/cuda/lib64 路径下要存在 链接到libcudnn.so.5.1.10 的libcudnn.so cudnn7系列只需要在/usr/local/cuda/lib64 路径下存在 libcudnn.so.7 就可以了,所以不会产生冲突这样就可以完成 多版本cudnn 并存...原创 2018-08-03 16:15:36 · 473 阅读 · 0 评论 -
hourglass pytorch 实现
主要分为几块1、数据集读取2、hg-model3、training翻译 2018-07-13 14:30:50 · 6646 阅读 · 17 评论