keras搭建神经网络（lenet）

keras搭建神经网络（lenet）

原文来源：https://blog.csdn.net/zuliang001/article/details/83269410

 

keras介绍

Keras 是一个用 Python 编写的高级神经网络 API，它能够以 TensorFlow, CNTK, 或者 Theano 作为后端运行。Keras 的开发重点是支持快速的实验。能够以最小的时延把你的想法转换为实验结果，是做好研究的关键。

论文链接

http://yann.lecun.com/exdb/publis/pdf/lecun-01a.pdf

keras代码实现

用的是28*28的MNIST数据集。

1. 头文件

#coding=utf-8

from keras.models import Sequential

from keras.layers import Dense,Flatten

from keras.layers.convolutional import Conv2D, MaxPooling2D

from keras.utils.np_utils import to_categorical

from keras.datasets import mnist

from keras.utils import plot_model

Sequential:顺序模型 

Dense：全连接，简称FC 

Flatten：上图中s4到c5的过程，相当于把16*5*5的feature map展开成400的特征向量，在通过全连接压成120维的特征向量 

Conv2D：2d卷积 

MaxPooling2D：2d下采样，文章中的subsampling 

to_categorical：把一维的向量转换为num_class维的One-hot编码 

from keras.datasets import mnist：keras自带了MNIST数据集 

plot_model：打印我们等下建好的模型，相当于可视化模型

2. 加载数据集