tensorflow实现简单的卷积网络


Posted in Python onMay 24, 2018

使用tensorflow实现一个简单的卷积神经,使用的数据集是MNIST,本节将使用两个卷积层加一个全连接层,构建一个简单有代表性的卷积网络。

代码是按照书上的敲的,第一步就是导入数据库,设置节点的初始值,Tf.nn.conv2d是tensorflow中的2维卷积,参数x是输入,W是卷积的参数,比如【5,5,1,32】,前面两个数字代表卷积核的尺寸,第三个数字代表有几个通道,比如灰度图是1,彩色图是3.最后一个代表卷积的数量,总的实现代码如下:

from tensorflow.examples.tutorials.mnist import input_data 
import tensorflow as tf 
mnist = input_data.read_data_sets("MNSIT_data/", one_hot=True) 
sess = tf.InteractiveSession() 
 
 
# In[2]: 
#由于W和b在各层中均要用到,先定义乘函数。 
#tf.truncated_normal:截断正态分布,即限制范围的正态分布 
def weight_variable(shape): 
  initial = tf.truncated_normal(shape, stddev=0.1) 
  return tf.Variable(initial) 
 
 
# In[7]: 
#bias初始化值0.1. 
def bias_variable(shape): 
  initial = tf.constant(0.1, shape=shape) 
  return tf.Variable(initial) 
 
 
# In[12]: 
#tf.nn.conv2d:二维的卷积 
#conv2d(input, filter, strides, padding, use_cudnn_on_gpu=None,data_format=None, name=None) 
#filter:A 4-D tensor of shape 
#   `[filter_height, filter_width, in_channels, out_channels]` 
#strides:步长,都是1表示所有点都不会被遗漏。1-D 4值,表示每歌dim的移动步长。 
# padding:边界的处理方式,“SAME"、"VALID”可选 
def conv2d(x, W): 
  return tf.nn.conv2d(x, W, strides=[1, 1, 1, 1], padding='SAME') 
 
#tf.nn.max_pool:最大值池化函数,即求2*2区域的最大值,保留最显著的特征。 
#max_pool(value, ksize, strides, padding, data_format="NHWC", name=None) 
#ksize:池化窗口的尺寸 
#strides:[1,2,2,1]表示横竖方向步长为2 
def max_pool_2x2(x): 
  return tf.nn.max_pool(x, ksize=[1, 2, 2, 1], strides = [1, 2, 2, 1], padding='SAME') 
 
 
x = tf.placeholder(tf.float32, [None, 784]) 
y_ = tf.placeholder(tf.float32, [None, 10]) 
#tf.reshape:tensor的变形函数。 
#-1:样本数量不固定 
#28,28:新形状的shape 
#1:颜色通道数 
x_image = tf.reshape(x, [-1, 28, 28, 1]) 
 
 
#卷积层包含三部分:卷积计算、激活、池化 
#[5,5,1,32]表示卷积核的尺寸为5×5, 颜色通道为1, 有32个卷积核 
W_conv1 = weight_variable([5, 5, 1, 32]) 
b_conv1 = bias_variable([32]) 
h_conv1 = tf.nn.relu(conv2d(x_image, W_conv1) + b_conv1) 
h_pool1 = max_pool_2x2(h_conv1) 
 
 
W_conv2 = weight_variable([5, 5, 32, 64]) 
b_conv2 = bias_variable([64]) 
h_conv2 = tf.nn.relu(conv2d(h_pool1, W_conv2) + b_conv2) 
h_pool2 = max_pool_2x2(h_conv2) 
 
 
#经过2次2×2的池化后,图像的尺寸变为7×7,第二个卷积层有64个卷积核,生成64类特征,因此,卷积最后输出为7×7×64. 
#tensor进入全连接层之前,先将64张二维图像变形为1维图像,便于计算。 
W_fc1 = weight_variable([7*7*64, 1024]) 
b_fc1 = bias_variable([1024]) 
h_pool2_flat = tf.reshape(h_pool2, [-1, 7*7*64]) 
h_fc1 = tf.nn.relu(tf.matmul(h_pool2_flat, W_fc1) + b_fc1) 
 
 
#对全连接层做dropot 
keep_prob = tf.placeholder(tf.float32) 
h_fc1_dropout = tf.nn.dropout(h_fc1, keep_prob) 
 
 
#又一个全连接后foftmax分类 
W_fc2 = weight_variable([1024, 10]) 
b_fc2 = bias_variable([10]) 
y_conv = tf.nn.softmax(tf.matmul(h_fc1_dropout, W_fc2) + b_fc2) 
 
 
cross_entropy = tf.reduce_mean(-tf.reduce_sum(y_*tf.log(y_conv), reduction_indices=[1])) 
#AdamOptimizer:Adam优化函数 
train_step = tf.train.AdamOptimizer(1e-4).minimize(cross_entropy) 
 
 
 
correct_prediction = tf.equal(tf.argmax(y_, 1), tf.argmax(y_conv, 1)) 
accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32)) 
 
 
#训练,并且每100个batch计算一次精度 
tf.global_variables_initializer().run() 
for i in range(20000): 
  batch = mnist.train.next_batch(50) 
  if i%100 == 0: 
    train_accuracy = accuracy.eval(feed_dict={x:batch[0], y_:batch[1], keep_prob:1.0}) 
    print("step %d, training accuracy %g" %(i, train_accuracy)) 
  train_step.run(feed_dict={x:batch[0], y_:batch[1], keep_prob:0.5}) 
 
 
#在测试集上测试 
print("test accuracy %g"%accuracy.eval(feed_dict={x:mnist.test.images, y_:mnist.test.labels, keep_prob:1.0}))

 注意的是书上开始运行的代码是tf.global_variables_initializer().run(),但是在敲到代码中就会报错,也不知道为什么,可能是因为版本的问题吧,上网搜了一下,改为sess.run(tf.initialiaze_all_variables)即可。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
将Python中的数据存储到系统本地的简单方法
Apr 11 Python
Python调用C语言的方法【基于ctypes模块】
Jan 22 Python
Python3实现的简单验证码识别功能示例
May 02 Python
selenium+python 对输入框的输入处理方法
Oct 11 Python
Python安装pycurl失败的解决方法
Oct 15 Python
Python PIL读取的图像发生自动旋转的实现方法
Jul 05 Python
Django打印出在数据库中执行的语句问题
Jul 25 Python
Python 使用list和tuple+条件判断详解
Jul 30 Python
利用python实现周期财务统计可视化
Aug 25 Python
python从zip中删除指定后缀文件(推荐)
Dec 05 Python
Python使用graphviz画流程图过程解析
Mar 31 Python
Python必备技巧之函数的使用详解
Apr 04 Python
解决pandas 作图无法显示中文的问题
May 24 #Python
TensorFlow实现简单卷积神经网络
May 24 #Python
解决matplotlib库show()方法不显示图片的问题
May 24 #Python
解决pandas无法在pycharm中使用plot()方法显示图像的问题
May 24 #Python
解决seaborn在pycharm中绘图不出图的问题
May 24 #Python
快速解决PyCharm无法引用matplotlib的问题
May 24 #Python
Django rest framework实现分页的示例
May 24 #Python
You might like
十天学会php之第四天
2006/10/09 PHP
为IP查询添加GOOGLE地图功能的代码
2010/08/08 PHP
PHP 类相关函数的使用详解
2013/05/10 PHP
PHP中func_get_args(),func_get_arg(),func_num_args()的区别
2013/09/30 PHP
用PHP来计算某个目录大小的方法
2014/04/01 PHP
PHP实现阳历到农历转换的类实例
2015/03/07 PHP
js 复制或插入Html的实现方法小结
2010/05/19 Javascript
找出字符串中出现次数最多的字母和出现次数精简版
2012/11/07 Javascript
JS常用正则表达式总结
2013/11/12 Javascript
jQuery+CSS3实现树叶飘落特效
2015/02/01 Javascript
js实现鼠标经过表格行变色的方法
2015/05/12 Javascript
JavaScript 数组some()和filter()的用法及区别
2016/05/20 Javascript
js自调用匿名函数的三种写法(推荐)
2016/08/19 Javascript
Node.js中的require.resolve方法使用简介
2017/04/23 Javascript
webstorm中vue语法的支持详解
2018/05/09 Javascript
详解Vue中watch的详细用法
2018/11/28 Javascript
Angular封装搜索框组件操作示例
2019/04/25 Javascript
webpack打包优化的几个方法总结
2020/02/10 Javascript
jQuery实现本地存储
2020/12/22 jQuery
把大数据数字口语化(python与js)两种实现
2013/02/21 Python
Python 正则表达式入门(初级篇)
2016/12/07 Python
浅析Python中return和finally共同挖的坑
2017/08/18 Python
python实现拓扑排序的基本教程
2018/03/11 Python
python实现求解列表中元素的排列和组合问题
2018/03/15 Python
python中eval与int的区别浅析
2019/08/11 Python
Python 线程池用法简单示例
2019/10/02 Python
opencv 图像滤波(均值,方框,高斯,中值)
2020/07/08 Python
Python实现给PDF添加水印的方法
2021/01/25 Python
澳大利亚网上书店:QBD
2021/01/09 全球购物
若干个Java基础面试题
2015/05/19 面试题
高中生职业规划范文
2014/03/09 职场文书
监督检查工作方案
2014/05/28 职场文书
一般基层干部群众路线教育实践活动个人对照检查材料
2014/11/04 职场文书
2014年幼儿园小班工作总结
2014/12/04 职场文书
海弦WR-800F
2022/04/05 无线电
golang的文件创建及读写操作
2022/04/14 Golang