神经网络 tensorflow教程 2.2 下载MNIST 数据集 (自动版)

文章参考:   https://zhuanlan.zhihu.com/p/25934529


环境:

语言 :python3.5(使用 Anaconda3-4.2.0-Windows-x86_64.exe)

操作系统: windos7


创建python 文件并执行   自动将数据集下载并展示前100条

#coding:utf-8
import tensorflow.examples.tutorials.mnist.input_data as input_data
import numpy as np
import matplotlib.pyplot as plt
import pylab

print (input_data)
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

batch_xs, batch_ys = mnist.train.next_batch(100)   #只取其中100条数据来看
for one_pic_vic in batch_xs:
    one_pic_arr = np.reshape(one_pic_vic,(28,28))
    pic_matrix = np.matrix(one_pic_arr,dtype = "float")
    plt.imshow(pic_matrix)
    pylab.show()



如果报错: 原因是  网络连接超时   则手动下载数据集,并在上述的py文件同级目录下创建文件夹MNIST_data。:

数据集下载地址:   http://yann.lecun.com/exdb/mnist/

下载对应的四个文件   :   t10k-images.idx3-ubyte       t10k-labels.idx1-ubyte      train-images.idx3-ubyte      train-labels.idx1-ubyte    


将数据集放到MNIST_data 下  执行 py 程序。 




综上: 

则会显示:







  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是使用TensorFlow实现卷积神经网络MNIST数据集进行分类的代码: ```python import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data # 加载MNIST数据集 mnist = input_data.read_data_sets("MNIST_data/", one_hot=True) # 定义占位符 x = tf.placeholder(tf.float32, [None, 784]) y_ = tf.placeholder(tf.float32, [None, 10]) # 将输入数据转换为图像格式 x_image = tf.reshape(x, [-1,28,28,1]) # 定义卷积层1 W_conv1 = tf.Variable(tf.truncated_normal([5, 5, 1, 32], stddev=0.1)) b_conv1 = tf.Variable(tf.constant(0.1, shape=[32])) h_conv1 = tf.nn.relu(tf.nn.conv2d(x_image, W_conv1, strides=[1, 1, 1, 1], padding='SAME') + b_conv1) # 定义池化层1 h_pool1 = tf.nn.max_pool(h_conv1, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME') # 定义卷积层2 W_conv2 = tf.Variable(tf.truncated_normal([5, 5, 32, 64], stddev=0.1)) b_conv2 = tf.Variable(tf.constant(0.1, shape=[64])) h_conv2 = tf.nn.relu(tf.nn.conv2d(h_pool1, W_conv2, strides=[1, 1, 1, 1], padding='SAME') + b_conv2) # 定义池化层2 h_pool2 = tf.nn.max_pool(h_conv2, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME') # 展开池化层2的输出,作为全连接层的输入 h_pool2_flat = tf.reshape(h_pool2, [-1, 7*7*64]) # 定义全连接层 W_fc1 = tf.Variable(tf.truncated_normal([7*7*64, 1024], stddev=0.1)) b_fc1 = tf.Variable(tf.constant(0.1, shape=[1024])) h_fc1 = tf.nn.relu(tf.matmul(h_pool2_flat, W_fc1) + b_fc1) # 定义输出层 W_fc2 = tf.Variable(tf.truncated_normal([1024, 10], stddev=0.1)) b_fc2 = tf.Variable(tf.constant(0.1, shape=[10])) y_conv = tf.matmul(h_fc1, W_fc2) + b_fc2 # 定义损失函数 cross_entropy = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(labels=y_, logits=y_conv)) # 定义优化器 train_step = tf.train.AdamOptimizer(1e-4).minimize(cross_entropy) # 定义评测准确率的操作 correct_prediction = tf.equal(tf.argmax(y_conv,1), tf.argmax(y_,1)) accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32)) # 开始训练 with tf.Session() as sess: sess.run(tf.global_variables_initializer()) 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]}) print('step %d, training accuracy %g' % (i, train_accuracy)) train_step.run(feed_dict={x: batch[0], y_: batch[1]}) print('test accuracy %g' % accuracy.eval(feed_dict={x: mnist.test.images, y_: mnist.test.labels})) ``` 这段代码中,我们定义了两个占位符x和y_,分别表示输入数据和标签。接着将输入数据转换为图像格式,并定义了两个卷积层和两个池化层,最后是一个全连接层和一个输出层。训练过程中,我们使用Adam优化器进行参数更新,并输出训练准确率和测试准确率。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值