数据集
mnist:6万张手写数字图片,大小为28 28
1(灰度图),10个类别
网络参数
- 输入神经元个数:
- 输出神经元个数:10
INPUT_NODE = 784
OUTPUT_NODE = 10
IMAGE_SIZE = 28
NUM_CHANNELS = 1
NUM_LABELS = 10
- 第一层:卷积层
使用边长为5,深度为32的卷积核,卷积核的移动步幅为1,且使用0填充
输出的尺寸为:28×28×32的张量
# 第一层卷积层的尺寸和深度
CONV1_DEEP = 32
CONV1_SIZE = 5
- 第二层:池化层
最大池化层卷积核边长为2,使用0填充,移动步幅为2.
输入为28×28×32的张量,输出为14×14×32的张量
网友评论