LeNet-5

作者: 蜉蝣之翼 | 来源:发表于2019-02-16 16:04 被阅读0次

数据集

mnist:6万张手写数字图片,大小为28 \times28 \times1(灰度图),10个类别

网络参数

  • 输入神经元个数:28\times 28\times1=784
  • 输出神经元个数: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的张量

相关文章

网友评论

      本文标题:LeNet-5

      本文链接:https://www.haomeiwen.com/subject/edtieqtx.html