- 博客(10)
- 资源 (27)
- 收藏
- 关注
原创 PyTorch之—图像分类一(每个类对应一个文件夹)
一、数据集准备数据按文件夹组织,每个类对应一个文件夹。大多数其他的PyTorch教程和示例都希望你先按照训练集和验证集来组织文件夹,然后在训练集和验证集中再按照类别进行组织。但我认为这非常麻烦,必须从每个类别中选择一定数量的图像并将它们从训练集文件夹移动到验证集文件夹。由于大多数人会通过选择一组连续的文件作为验证集,因此选择可能存在很多偏差。因此,这儿有一个将数据集快速分为训练集和测试集的更好...
2019-02-27 17:29:27 9121 1
原创 PyTorch之一Fine-tuning / hook
文章目录一、Fine tuning(微调)二、计算机视觉方向的微调三、案例:一、Fine tuning(微调)针对于某个任务,自己的训练数据不多,那怎么办? 没关系,我们先找到一个同类的别人训练好的模型,把别人现成的训练好了的模型拿过来,换成自己的数据,调整一下参数,在训练一遍,(此后的训练需要冻结之前网络层的参数,训练之后添加的网络层参数)这就是微调(fine-tune)。PyTorch里...
2019-02-24 11:10:42 1291
原创 PyTorch—数据加载器
我们之前示范的许多模型示例使用的数据集都是比较少的,但是一旦涉及到需要加载大量的数据集时,会面临内存不够用的局面,这里就涉及到了大数据集的加载问题。大数据量数据量:将所有的图像数据直接加载到numpy数据中会占用大量的内存由于需要对数据进行导入,每次训练的时候在数据读取阶段会占用大量的时间只使用了单线程去读取,读取效率比较低下拓展性很差,如果需要对数据进行一些预处理,例如数据增强时会消...
2019-02-21 20:20:42 1404
原创 PyTorch之四—梯度下降 和 关系拟合
文章目录PyTorch梯度传递线性模型线性回归示例PyTorch梯度传递在PyTorch中,传入网络计算的数据类型必须是 Variable 类型, Variable包装了一个Tensor,并且保存着梯度和创建这个Variable function的引用,换句话说,就是记录网络每层的梯度和网络图,可以实现梯度的反向传递。整体网络图如下:input→conv2d→relu→maxpool2d→...
2019-02-21 19:07:30 745
原创 PyTorch之三—动态图
文章目录一、概述二、Torch三、动态图效果一、概述目前神经网络框架分为静态图框架和动态图框架,PyTorch 和 TensorFlow、Caffe 等框架最大的区别就是他们拥有不同的计算图表现形式。TensorFlow 使用静态图,这意味着我们先定义计算图,然后不断使用它,中间是不能够改变它的计算图的,且定义静态图时需要使用新的特殊语法,这也意味着图设定时无法使用if、while、for-...
2019-02-21 19:06:06 2888 1
原创 PyTorch之—可视化(Visdom/TensorboardX)
文章目录一、Visdom安装与使用小案例二、TensorBoardX一、VisdomVisdom是Facebook在2017年发布的一款针对PyTorch的可视化工具。visdom由于其功能简单,一般会被定义为服务器端的matplot,也就是说我们可以直接使用python的控制台模式进行开发并在服务器上执行,将一些可视化的数据传送到Visdom服务上,通过Visdom服务进行可视化。官网:h...
2019-02-20 09:21:04 13211
原创 TensorFlow之九—损失函数
文章目录损失函数(loss)TensorFlow损失函数损失函数(loss)用来表示预测(y)与已知答案(y_)的差距。在训练神经网络时,通过不断改变神经网络中所有参数,使损失函数不断减小,从而训练出更高准确率的神经网络模型,常用的损失函数有均方误差,自定义和交叉熵等。TensorFlow损失函数TensorFlow内置了四个损失函数:Tensor=tf.nn.softmax_cros...
2019-02-19 09:23:41 572
原创 PyTorch之八—优化器&学习率
文章目录torch.optim 优化器使用`torch.optim`:进行单次优化 optimizer.step()使用闭包优化 optimizer.step(closure)常用的优化器函数学习率 与学习率调节torch.optim 优化器使用torch.optim:构建一个optimizer对象。这个对象能够保持当前参数状态并基于计算得到的梯度进行参数更新。你需要给它一个包含了需要优...
2019-02-15 20:49:41 4958 1
原创 PyTorch之七—MNIST 手写数字识别
本节基于MNIST数据集,实现CNN学习过程。import osimport torchimport torch.nn as nnfrom torch.autograd import Variableimport torch.utils.data as Dataimport torchvisionimport matplotlib.pyplot as plttorch.manu...
2019-02-15 15:35:04 7728 3
原创 TensorFlow之八—优化器
文章目录一,优化器函数tf.train.GradientDescentOptimizer(梯度下降算法)tf.train.MomentumOptimizer(动量梯度下降算法 )tf.train.AdamOptimizer(Adam优化算法)一,优化器函数TensorFlow中提供了如下优化器函数:tf.train.Optimizertf.train.GradientDescentOpt...
2019-02-12 14:12:17 637
goturn-files.7z
2021-09-17
google_class.7z
2021-09-17
face_detector_检测模型.7z
2021-09-17
华为通用表格识别格式化成excel
2020-03-20
LC算法python实现.zip
2019-11-08
psf2otf_circShift函数.zip
2019-07-23
基于RFM模型实现的零售精准营销响应预测系统代码
2019-02-15
基于RFM模型实现的零售精准营销响应预测系统数据
2018-11-01
文本识别opencv-text-recognition
2018-10-16
工业生产流程的数据分析预测
2018-08-29
基于聚类(Kmeans)算法实现客户价值分析系统(电信运营商)
2018-08-20
基于聚类(Kmeans)算法实现的客户价值分析系统data
2018-08-18
基于NLP自然语言构建的文档自动分类_test_data _stopwords
2018-08-18
从新闻网站点击流中挖掘
2018-08-15
卷积神经网络实现数字识别数据
2018-08-14
基于Kaggle的经典AI项目——数据集
2018-08-09
shape_predictor_68_face_landmarks和dlib
2018-08-08
pyspark_ml_pipeline_DecisionTreeClassifier_RF数据集
2015-06-29
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人