- 博客(4)
- 收藏
- 关注
原创 关于安装tensoflow-gpu版本的几个坑
首先是配置,window7旗舰版+gtx765m+8G内存,网上大多数安装教程都是基于win10的,win7遗老不多了.下面是安装步骤:首先是查看你的gpu是否支持CUDA,点击链接,按照官方的说法是计算能力大于3.0的才适合使用gpu版本,我的显卡刚好合适.下面是安装CUDA和cuDNN,这里一定要注意版本之间的搭配,假如你的tensorflow是1.8的,对应的CUDA就是9.0,对应...
2018-08-16 16:48:39
1217
原创 基于cnn的图像二分类算法(二)
上一篇我们简单介绍了使用卷积神经网络进行图片分类的算法,点这里,然而最后的效果并不理想,现在我们来考虑如何改进这个算法.这次加入的功能是保存参数的功能.我们训练神经网络,希望损失值越低越好,test的时候希望精确率越高越好.然而最终的目的还是对一张或者几张图片进行分类,所以我们需要保存训练的参数,tensorflow很方便就有这样的功能.tf.train.Saver()函数就是用来保...
2018-08-03 17:32:41
5907
2
原创 基于keras的数据增广(data augmentation)
在实际的深度学习当中,我们通常会遇到收集的数据过少,收集数据过于困难等,导致训练过程中很容易发生过拟合等现象.这时候我们就需要对数据进行扩充来避免这种现象.下面就以图片来做例子,最简单的对图片进行扩充的方法是翻转,包括水平翻转和垂直翻转,但一般图像用到最多的是水平翻转,(正常物体不可能跑到天上去).还有对图像进行扭曲,变形,拉伸,填充,改变hsv或者rgb的值等等.keras数据库提供了很...
2018-08-02 18:01:57
2955
1
原创 基于cnn的图像二分类算法(一)
本算法是基于tensorflow,使用python语言进行的一种图像分类算法,参考于谷歌的mnist手写识别,包括以下几个模块:图像读取,图像处理,图像增强。卷积神经网络部分包括:卷积层1,汇合层1(部分文献也有叫池化层的),卷积层2,汇合层2,全连接层1,全连接层2,共6层神经网络。损失函数采用交叉熵,优化则采用adam优化法,由于数据集大小较小,只有200张图片,故没有采用MBGD梯度下降算法...
2018-07-27 14:34:54
23779
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人