前言

这是一个可以真正跑下来的全连接神经网络识别手写数字问题的代码哟，又不懂的语句或者逻辑，欢迎评论区留言

代码

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers,optimizers,datasets#数据预处理
(x,y),(x_val,y_val) = datasets.mnist.load_data()
x = tf.convert_to_tensor(x,dtype=tf.float32)/255.
y = tf.convert_to_tensor(y,dtype=tf.int32)
y = tf.one_hot(y,depth=10)
print(x.shape,y.shape)
train_dateset = tf.data.Dataset.from_tensor_slices((x,y))
train_dateset = train_dateset.batch(200)#建立神经网络模型
model = keras.Sequential([layers.Dense(512,activation='relu'),layers.Dense(256,activation='relu'),layers.Dense(128,activation='relu'),layers.Dense(64,activation='relu'),layers.Dense(10)]
)optimizers = optimizers.SGD(learning_rate=0.001)#定义训练流程
def train_epoch(epoch):for step,(x,y) in enumerate(train_dateset):with tf.GradientTape() as tape:x = tf.reshape(x,(-1,28*28))out = model(x)loss = tf.reduce_sum(tf.square(out-y))/x.shape[0]grads = tape.gradient(loss,model.trainable_variables)optimizers.apply_gradients(zip(grads,model.trainable_variables))if step%100 == 0:print(epoch,step,'loss',loss.numpy())#运行
def train():for epoch in range(30):train_epoch(epoch)if __name__ == '__main__':train()

TensorFlow实现mnist手写数字识别项目代码相关推荐

mnist手写数字识别python_Python tensorflow实现mnist手写数字识别示例【非卷积与卷积实现】...
本文实例讲述了Python tensorflow实现mnist手写数字识别.分享给大家供大家参考,具体如下: 非卷积实现 import tensorflow as tf from tensorflow ...
基于CNN的MINIST手写数字识别项目代码以及原理详解
文章目录项目简介项目下载地址项目开发软件环境项目开发硬件环境前言一.数据加载的作用二.Pytorch进行数据加载所需工具 2.1 Dataset 2.2 Dataloader 2.3 T ...
mnist手写数字识别python_基于tensorflow的MNIST手写数字识别（二）--入门篇
一.本文的意义因为谷歌官方其实已经写了MNIST入门和深入两篇教程了,那我写这些文章又是为什么呢,只是抄袭?那倒并不是,更准确的说应该是笔记吧,然后用更通俗的语言来解释,并且补充更多,官方文章中没有 ...
【TensorFlow】MNIST手写数字识别
MNIST MNIST是一个非常简单的机器视觉数据集.如图,它由几万字28像素×28像素的手写数字组成,这些图片只包含灰度值信息.我们的任务是对这些手写数字的图片进行分类,转成0~9一共10类. ...
手写数字识别项目代码——卷积神经网络LeNet-5模型
''' #2018-06-25 272015 June Monday the 26 week, the 176 day SZ 手写字体识别程序文件1: 这个程序使用了卷积神经网络LeNet - 5模型 ...
深度学习第一周 tensorflow实现mnist手写数字识别
tensorflow saver_机器学习入门(6)：Tensorflow项目Mnist手写数字识别-分析详解
本文主要内容:Ubuntu下基于Tensorflow的Mnist手写数字识别的实现训练数据和测试数据资料:http://yann.lecun.com/exdb/mnist/ 前面环境都搭建好了,直接 ...
基于K210的MNIST手写数字识别
基于K210的MNIST手写数字识别项目已开源链接: Github. 硬件平台采用Maixduino开发板在sipeed官方有售软件平台使用MaixPy环境进行单片机的编程官方资源可在这里 ...
MNIST手写数字识别【Matlab神经网络工具箱】
MNIST手写数字识别 Matlab代码: %Neural Networks Codes will be run on this part tic %%%%%%%%%%%%%%%%%%%%%%%%%% ...
将tensorflow训练好的模型移植到Android (MNIST手写数字识别)
将tensorflow训练好的模型移植到Android (MNIST手写数字识别) [尊重原创,转载请注明出处]https://blog.csdn.net/guyuealian/article/det ...

TensorFlow实现mnist手写数字识别项目代码

前言

代码

TensorFlow实现mnist手写数字识别项目代码相关推荐

最新文章

热门文章