本文使用的数据集是MNIST,主要使用两个卷积层加一个全连接层构建的卷积神经网络。
先载入MNIST数据集(手写数字识别集),并创建默认的Interactive Session(在没有指定回话对象的情况下运行变量)
from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf
mnist = input_data.read_data_sets(MNIST_data/, one_hot=True)
sess = tf.InteractiveSession()
在定义一个初始化函数,因为卷
2024-03-17 17:06:44
80KB
mnist
1