核大小:5x5
核数目:6
步长:1
输出特征图大小:6x28x28
激活函数采用sigmoid
采用平均池化
核大小:2x2
步长:2
输出特征图大小:6x14x14
核大小:5x5
核数目:16
步长:1
输出特征图大小:16x10x10
激活函数采用sigmoid
采用平均池化
核大小:2x2
步长:2
输出特征图大小:16x5x5
输入:400维
输出:120维
输入:120维
输出:84维
输入:84维
输出:10维
代码部分
LeNet5.py定义了网络结构
train.py训练模型,生成onnx文件。可以通过Neutron查看网络结构。
代码见github



