基于Tensorflow的MNIST手写数字识别分类

最新推荐文章于 2021-10-20 15:27:24 发布

qq_40579095

最新推荐文章于 2021-10-20 15:27:24 发布

阅读量310

点赞数 1

文章标签： tensorflow mnist

本文链接：https://blog.csdn.net/qq_40579095/article/details/88804019

版权

本文介绍如何使用Tensorflow框架进行MNIST手写数字数据集的分类任务，通过示例代码详细展示了模型训练过程。

摘要由CSDN通过智能技术生成

代码如下：

import tensorflow as tf
import numpy as np
from tensorflow.examples.tutorials.mnist import input_data
from tensorflow.contrib.tensorboard.plugins import projector
import time

IMAGE_PIXELS = 28
hidden_unit = 100
output_nums = 10
learning_rate = 0.001
train_steps = 50000
batch_size = 500
test_data_size = 10000
#日志目录(这里根据自己的目录修改)
logdir = 'D:/Develop_Software/Anaconda3/WorkDirectory/summary/mnist'
#导入mnist数据
mnist = input_data.read_data_sets('MNIST_data', one_hot = True)

  #全局训练步数
global_step = tf.Variable(0, name = 'global_step', trainable = False)
with tf.name_scope('input'):
  #输入数据
  with tf.name_scope('x'):
    x = tf.placeholder(
        dtype = tf.float32, shape = (None, IMAGE_PIXELS * IMAGE_PIXELS))
  #收集x图像的会总数据
  with tf.name_scope('x_summary'):
    shaped_image_batch = tf.reshape(
        tensor = x,
        shape = (-1, IMAGE_PIXELS, IMAGE_PIXELS, 1),
        name = 'shaped_image_batch')
    tf.summary.image(name = 'image_summary',
                     tensor = shaped_image_batch,
                     max_outputs = 10)
  with tf.name_scope('y_'):
    y_ = tf.placeholder(dtype = tf.float32, shape = (None