- 博客(10)
- 资源 (1)
- 收藏
- 关注
原创 模型性能指标
这里首先介绍几个常见的模型评价术语,现在假设我们进行二分类的任务,可将样例根据其真实类别与学习器预测类别的组合划分为真正例(true positive)、假正例(false positive)、真反例(true negative)、假反例(false negative)四种情形:1. True positives(TP): 被正确地划分为正例的个数,即实际为正例且被分类器划分为正例的样本数;...
2019-08-26 14:38:28 1678
转载 PyTorch 常用代码段整理合集
https://zhuanlan.zhihu.com/p/59205847本文代码基于PyTorch 1.0版本,需要用到以下包import collectionsimport osimport shutilimport tqdmimport numpy as npimport PIL.Imageimport torchimport torchvision1. 基础配...
2019-08-26 10:39:17 2918 1
转载 全卷积网络FCN
背景CNN能够对图片进行分类,可是怎么样才能识别图片中特定部分的物体,在2015年之前还是一个世界难题。神经网络大神Jonathan Long发表了《Fully Convolutional Networks for Semantic Segmentation》在图像语义分割挖了一个坑,于是无穷无尽的人往坑里面跳。全卷积网络 Fully Convolutional NetworksC...
2019-08-25 17:04:43 898
原创 卷积类型
https://baijiahao.baidu.com/s?id=1625255860317955368&wfr=spider&for=pchttps://blog.csdn.net/ahxieqi/article/details/93628533
2019-08-25 17:02:47 192
转载 使用Lua CJSON库进行encode与decode操作完成对Json数据转化
https://blog.csdn.net/vivi_12/article/details/76549066
2019-08-16 16:57:45 1502
原创 Pytorch Visdom可视化工具
Pytorch官方支持的可视化工具是Visdom(当然也支持TensorBoardX),Visdom更简洁方便一些(例如对image数据的可视化可以直接使用Tensor,而不必转到cpu上再转为numpy数据),刷新率也更快。1. 安装Visdomsudo pip3 install visdom2. 启动Visdom服务python3 -m visdom.server启动成...
2019-08-07 17:24:38 776
原创 Pytorch 多分类结果测试
在模型训练过程中需要对当前的效果进行验证,或者训练结束后需要在测试集上对模型进行测试。比如多分类问题,网络的前向传播的结果是一个概率值Tensor,如果是一个10分类问题,并且batch=4,结果是一个4*10的Tensor,Tensor的每一行表示某张图片分别在10分类下的预测概率值。Pytorch中的argmax()函数可以返回Tensor中每一行最大值的索引,torch.eq()函数可以...
2019-08-07 16:51:37 5021
原创 Pytorch 使用GPU加速
Pytorch中使用torch.device()选取并返回抽象出的设备,然后在定义的网络模块或者Tensor后面加上.to(device变量)就可以将它们搬到设备上了。以上一篇代码为例,使用GPU设备:device = torch.device('cuda:0') # 使用第一张显卡需要将如下部分搬移到GPU上:1. 定义的网络net = MLP().to(device)...
2019-08-07 16:09:39 1940
原创 Pytorch 不同的网络参数初始化对梯度消失的影响
尽管ReLU相比其它激活函数已经大大避免了发生梯度消失或者为0的现象,但不好的参数初始化方法也会导致这种情况的出现。1. 使用标准正态分布初始化这里用torch.randn()来做实际上就是对参数w和b进行了N~(0,1)的标准正态分布初始化,这个初始化值都很小,比较接近0。import torchfrom torch.nn import functional as Ffrom...
2019-08-07 15:51:32 1588
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人