随笔分类 -  pytorch

[项目实战］训练retinanet(pytorch版）
摘要：pytorch, retinanet,目标检测 阅读全文
posted @ 2019-05-08 17:48 wuzeyuan 阅读(9050) 评论(1) 推荐(2)
【语义分割】PSPNet中PSP模块的pytorch实现
摘要：github地址：https://github.com/Lextal/pspnet-pytorch/blob/master/pspnet.py PSP模块示意图如下 代码如下 此外，我基于自己的工作稍加修改，也给出一个3D版本。改动有几处，一是3d卷积和池化，二是上采样由双线性插值切换为trilin 阅读全文
posted @ 2019-01-05 15:52 wuzeyuan 阅读(7433) 评论(1) 推荐(1)
【语义分割】large kernel matters中GCN模块的pytorch实现
摘要：GCN模块的实现比较简单，在giuhub上看到两种实现，轻微不同 实现一：https://github.com/ycszen/pytorch-segmentation/blob/master/gcn.py 实现二：https://github.com/ogvalt/large_kernel_matt 阅读全文
posted @ 2018-12-28 17:42 wuzeyuan 阅读(1580) 评论(0) 推荐(1)
对Deeplung检测+两样性分类的一个整合
摘要：整体的流程分为以下几步： 读取原始数据（.mhd文件）——> 生成mask ——> 对数据预处理 ——> 执行检测 ——> 对检测结果进行分类 ——>可视化 懒一点，不贴代码了，直接截图吧，反正就是记录一下。 可视化结果图 现在整合的这个东西以后要放到网上去，检测结果还好，分类效果一般。 阅读全文
posted @ 2018-11-21 20:38 wuzeyuan 阅读(2498) 评论(7) 推荐(3)
解读 pytorch对resnet的官方实现
摘要：pytorch resnet 阅读全文
posted @ 2018-10-30 23:26 wuzeyuan 阅读(93892) 评论(2) 推荐(13)
pytorch 迁移学习[摘自官网]
摘要：迁移学习包含两种：微调和特征提取器。 微调：对整个网络进行训练，更新所有参数 特征提取器：只对最后的输出层训练，其他层的权重保持不变 当然，二者的共性就是需要加载训练好的权重，比如在ImageNet上训练过的vgg,resnet等等。 那么，不管是微调还是特征提取器，大致都要遵从四个步骤。 阅读全文
posted @ 2018-10-23 16:14 wuzeyuan 阅读(865) 评论(0) 推荐(0)
pytorch 0.4.1安装问题
摘要：环境 ubuntu 16.04, anaconda3, python 2.7 将pytorch升级为0.4.1后，import torch报错 undefined symbol: _ZN4thpp10THCSTensorIlEC1EP8THCStateP14THCSLongTensor 在谷歌中查了 阅读全文
posted @ 2018-10-12 22:12 wuzeyuan 阅读(2953) 评论(0) 推荐(0)
pytorch 检测图片中是否有人
摘要：pytorch判断图片中有无人 阅读全文
posted @ 2018-10-06 14:36 wuzeyuan 阅读(2247) 评论(0) 推荐(1)
pytorch tensor与numpy转换
摘要：从官网拷贝过来的，就是做个学习记录。版本 0.4 tensor to numpy 输出 进行转换 输出 注意，转换后的tensor与numpy指向同一地址，所以，对一方的值改变另一方也随之改变 numpy to tensor 输出 除chartensor外所有tensor都可以转换为numpy 阅读全文
posted @ 2018-09-30 21:48 wuzeyuan 阅读(303628) 评论(2) 推荐(5)
pytorch实现squeezenet
摘要：pytorch,squeezenet 阅读全文
posted @ 2018-09-26 22:40 wuzeyuan 阅读(2506) 评论(1) 推荐(0)
pytorch实现depthwise convolution
摘要：depthwise convolution 阅读全文
posted @ 2018-09-22 14:35 wuzeyuan 阅读(17554) 评论(1) 推荐(2)
pytorch contiguous的使用
摘要：contiguous一般与transpose，permute,view搭配使用 即使用transpose或permute进行维度变换后，调用contiguous，然后方可使用view对维度进行变形。 具体原因我还没搞清，看到网上有两种说法，一种是维度变换后tensor在内存中不再是连续存储的，而vi 阅读全文
posted @ 2018-09-18 13:19 wuzeyuan 阅读(6053) 评论(0) 推荐(0)
pytorch 加载mnist数据集报错not gzip file
摘要：利用pytorch加载mnist数据集的代码如下 第一次使用mnist，需要下载，具体方法就是设置download=True，然而我运行的时候报错了。 错误：not gzip file 可是明明是.gz文件啊，查了几篇博客也没有说清楚原因的，于是自行下载了四个文件（训练集，测试集以及各自的标签），放 阅读全文
posted @ 2018-09-15 23:17 wuzeyuan 阅读(5999) 评论(2) 推荐(0)
[转载]pytorch自定义数据集
摘要：转自pytorch中文教程网 阅读全文
posted @ 2018-08-12 02:09 wuzeyuan 阅读(2314) 评论(0) 推荐(0)
[转载]什么情况下应该设置 cudnn.benchmark = True？
摘要：转自pytorch中文教程网 阅读全文
posted @ 2018-08-12 01:57 wuzeyuan 阅读(432) 评论(0) 推荐(0)
pytorch构建优化器
摘要：莫凡python学习笔记 阅读全文
posted @ 2018-08-11 16:15 wuzeyuan 阅读(2447) 评论(0) 推荐(0)
pytorch批训练数据构造
摘要：莫凡python学习笔记 阅读全文
posted @ 2018-08-11 14:43 wuzeyuan 阅读(782) 评论(0) 推荐(0)
pytorch搭建网络，保存参数，恢复参数
摘要：莫凡python学习笔记 阅读全文
posted @ 2018-08-10 23:37 wuzeyuan 阅读(1669) 评论(0) 推荐(0)